Merge pull request #46 from serengil/feat-task-2612-issue-30

serengil · web-flow · commit 9d5697ae98ca · 2023-12-26T14:50:08.000Z
no subdataset for else condition
diff --git a/chefboost/training/Training.py b/chefboost/training/Training.py
@@ -556,11 +556,8 @@ def buildDecisionTree(
 
     # ---------------------------
     # add else condition in the decision tree
-
     if df.Decision.dtypes == "object":  # classification
-        pivot = pd.DataFrame(subdataset.Decision.value_counts()).sort_values(
-            by=["count"], ascending=False
-        )
+        pivot = pd.DataFrame(df.Decision.value_counts()).sort_values(by=["count"], ascending=False)
         else_decision = f"return '{str(pivot.iloc[0].name)}'"
 
         if enableParallelism != True:
@@ -588,7 +585,7 @@ def buildDecisionTree(
             decision_rules.append(sample_rule)
 
     else:  # regression
-        else_decision = f"return {subdataset.Decision.mean()}"
+        else_decision = f"return {df.Decision.mean()}"
 
         if enableParallelism != True:
             functions.storeRule(file, (functions.formatRule(root), "else:"))
