Made so that the program runs

pull/1/head
E. Almqvist 4 years ago
parent 1493824ac7
commit 788ee4fb2d
  1. 7
      rgbAI/lib/func.py

@ -1,4 +1,5 @@
import numpy as np import numpy as np
from copy import deepcopy as copy
class AIlib: class AIlib:
def sigmoid(x): def sigmoid(x):
@ -38,8 +39,6 @@ class AIlib:
def propDer( dCost, dProp ): def propDer( dCost, dProp ):
# Calculate the partial derivative for that prop # Calculate the partial derivative for that prop
#print("################")
#print(dCost, dProp)
return dCost / dProp return dCost / dProp
def gradient( inp:np.array, obj, theta:float, maxLayer:int, layerIndex: int=0, grads=None, obj1=None, obj2=None ): # Calculate the gradient for that prop def gradient( inp:np.array, obj, theta:float, maxLayer:int, layerIndex: int=0, grads=None, obj1=None, obj2=None ): # Calculate the gradient for that prop
@ -49,8 +48,8 @@ class AIlib:
# Create new instances of the object # Create new instances of the object
if( not obj1 or not obj2 ): if( not obj1 or not obj2 ):
obj1 = obj obj1 = copy(obj) # annoying way to create a new instance of the object
obj2 = obj obj2 = copy(obj)
obj2.weights[layerIndex] += theta # mutate the second object obj2.weights[layerIndex] += theta # mutate the second object
obj2.bias[layerIndex] += theta obj2.bias[layerIndex] += theta

Loading…
Cancel
Save