From 9aeff11066447843edeced678ba8af9f221e72b3 Mon Sep 17 00:00:00 2001 From: Eleni Date: Fri, 15 Jan 2021 10:47:57 +0200 Subject: [PATCH] add sample size @ nodes --- Exareme-Docker/src/mip-algorithms/CART/cart_lib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Exareme-Docker/src/mip-algorithms/CART/cart_lib.py b/Exareme-Docker/src/mip-algorithms/CART/cart_lib.py index 85e596c01..225433122 100644 --- 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,