Gradient calculation

4 years ago · 80013cab71
parent ac6cb89bfe
commit 80013cab71
1 changed files with 5 additions and 5 deletions
--- a/rgbAI/lib/func.py
+++ b/rgbAI/lib/func.py
@ -53,12 +53,12 @@ class AIlib:

    def gradient( dCost:float, prop:list ):
        propLen = len(prop)
-        #print("PropLEN: ", propLen)
-        #print(prop)
-        #print("\n")
        gradient = [None] * propLen
-        for i in range( propLen ):
+        for i in range( propLen, 0, -1 ):
+            if( i == propLen ):
                gradient[i] = dCost / prop[i]
+            else:
+                gradient[i] = dCost / (prop[i] + gradient[i+1])

        return gradient