Changing colors to rainbow colors

examples/xor/sketch.js

@@ -1,7 +1,6 @@
 let nn;
 let lr_slider;
-
-let training_data = [{
+et training_data = [{
     inputs: [0, 0],
     outputs: [0]
   },
@@ -18,25 +17,20 @@ let training_data = [{
     outputs: [0]
   }
 ];
-
 function setup() {
   createCanvas(400, 400);
   nn = new NeuralNetwork(2, 4, 1);
   lr_slider = createSlider(0.01, 0.5, 0.1, 0.01);
-
+  colorMode(HSB);
 }
-
 function draw() {
   background(0);
-
   for (let i = 0; i < 10; i++) {
     let data = random(training_data);
     nn.train(data.inputs, data.outputs);
   }
-
   nn.setLearningRate(lr_slider.value());
-
-  let resolution = 10;
+	let resolution = 5;
   let cols = width / resolution;
   let rows = height / resolution;
   for (let i = 0; i < cols; i++) {
@@ -46,11 +40,8 @@ function draw() {
       let inputs = [x1, x2];
       let y = nn.predict(inputs);
       noStroke();
-      fill(y * 255);
+      fill(y * 360, 100, 100);
       rect(i * resolution, j * resolution, resolution, resolution);
     }
   }
-
-
-
 }