Fixed small bug

rgbAI/lib/func.py

@@ -87,7 +87,7 @@ class AIlib:
             obj.weights[i] -= obj.learningrate * gradient[i]["weight"] # mutate the weights
             obj.bias[i] -= obj.learningrate * gradient[i]["bias"]
 
-    def learn( inputNum:int, targetCost:float, obj, curCost: float=None ):
+    def learn( inputNum:int, targetCost:float, obj, theta:float, curCost: float=None ):
         # Calculate the derivative for:
         # Cost in respect to weights
         # Cost in respect to biases

rgbAI/main.py

@@ -29,7 +29,7 @@ class rgb(object):
         return cost
 
     def learn( self ):
-        ai.learn( 3, 0.001, self, 0.0001 )
+        ai.learn( 3, 0.001, self, 0.001 )
 
     def think( self, inp:np.array ):
         print("-----Gen " + str(self.generation) + "------")