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

no subdataset for else condition
serengil · Dec 26, 2023 · 9d5697a · 9d5697a
2 parents b54df75 + 9bf190d
commit 9d5697a
Showing 1 changed file with 2 additions and 5 deletions.
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:"))