Commit c79034c

HarshPathak310MaximSmolskiy

and

authored

Update logical issue in decision_tree.py (TheAlgorithms#13303)

Co-authored-by: Maxim Smolskiy <mithridatus@mail.ru>

1 parent 8edc478 commit c79034cCopy full SHA for c79034c

File tree

-4

lines changed

-4

lines changed

Lines changed: 3 additions & 4 deletions

Original file line number	Diff line number	Diff line change
`@@ -146,14 +146,13 @@ def predict(self, x):`
`146`	`146`	`"""`
`147`	`147`	`if self.prediction is not None:`
`148`	`148`	`return self.prediction`
`149`		`- elif self.left or self.right is not None:`
	`149`	`+ elif self.left isnotNoneand self.right is not None:`
`150`	`150`	`if x >= self.decision_boundary:`
`151`	`151`	`return self.right.predict(x)`
`152`	`152`	`else:`
`153`	`153`	`return self.left.predict(x)`
`154`	`154`	`else:`
`155`		`- print("Error: Decision tree not yet trained")`
`156`		`- return None`
	`155`	`+ raise ValueError("Decision tree not yet trained")`
`157`	`156`
`158`	`157`
`159`	`158`	`class TestDecisionTree:`
`@@ -201,4 +200,4 @@ def main():`
`201`	`200`	`main()`
`202`	`201`	`import doctest`
`203`	`202`
`204`		`- doctest.testmod(name="mean_squarred_error", verbose=True)`
	`203`	`+ doctest.testmod(name="mean_squared_error", verbose=True)`

Comments

(0)