TheAlgorithms/Python
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Update logical issue in decision_tree.py (#13303)

Browse Source 
Co-authored-by: Maxim Smolskiy <mithridatus@mail.ru>
This commit is contained in:
Harsh Pathak
2025-10-17 06:30:44 +05:30
committed by GitHub
parent 8edc478a19
commit c79034ca21
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
machine_learning/decision_tree.py
View File

@@ -146,14 +146,13 @@ class DecisionTree:
""" """
if self.prediction is not None: if self.prediction is not None:
return self.prediction return self.prediction
elif self.left or self.right is not None: elif self.left is not None and self.right is not None:
if x >= self.decision_boundary: if x >= self.decision_boundary:
return self.right.predict(x) return self.right.predict(x)
else: else:
return self.left.predict(x) return self.left.predict(x)
else: else:
print("Error: Decision tree not yet trained") raise ValueError("Decision tree not yet trained")
return None
class TestDecisionTree: class TestDecisionTree:
@@ -201,4 +200,4 @@ if __name__ == "__main__":
main() main()
import doctest import doctest
doctest.testmod(name="mean_squarred_error", verbose=True) doctest.testmod(name="mean_squared_error", verbose=True)