Update classifier.py

classifier.py

@@ -109,7 +109,7 @@ class Classifier:
                 # To simplify the query, we separated 
                 # the calculation into class numbers
                 grad_q_str = f"""
-                    SELECT sumForEachArray(arrayMap((x,y,gt)->arrayMap(i->i*(y-gt), x), X, Y, GT)) AS grad
+                    SELECT avgForEachArray(arrayMap((x,y,gt)->arrayMap(i->i*(y-gt), x), X, Y, GT)) AS grad
                     FROM (
                         SELECT groupArray(arrayPopBack(prelogit)) AS X, 
                             groupArray(1/(1+exp(-arraySum(arrayMap((x,y)->x*y, prelogit, {xq_s[n]}))))) AS Y, {labels} AS GT