add sample size @ nodes

madgik · Jan 15, 2021 · 9aeff11 · 9aeff11
1 parent 34b7fed
commit 9aeff11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Exareme-Docker/src/mip-algorithms/CART/cart_lib.py b/Exareme-Docker/src/mip-algorithms/CART/cart_lib.py
@@ -106,7 +106,7 @@ def tree_to_json(self):
                  "threshold" : None if self.gain == 0 or (self.right is None and self.left is None) else self.threshold,
                  "criterion" :self.criterion,
                  "gain" : self.gain,
-                 #"samples" : "Less than " + str(PRIVACY_MAGIC_NUMBER) if self.samples <= PRIVACY_MAGIC_NUMBER else self.samples,
+                 "samples" : "Less than " + str(PRIVACY_MAGIC_NUMBER) if self.samples <= PRIVACY_MAGIC_NUMBER else self.samples,
                  #"samplesPerClass" : samplesPerClass,
                  "classValue" : myclassValue,
                  "class" : myclass,