Skip to content

Commit 6057b13

Browse files
webmasterprowebmasterpro
authored andcommitted
updated decision tree with scoring functions
1 parent defe474 commit 6057b13

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

machine_learning/decision_tree.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,8 @@ def main():
136136
print("Average error: " + str(avg_error))
137137
print("Scoring mae: ") + str(scoring_functions.mae(test_cases,predictions))
138138
print("Scoring rmse: ") + str(scoring_functions.rmse(test_cases,predictions))
139+
print("Scoring rmsle: ") + str(scoring_functions.rmsle(test_cases,predictions))
140+
print("Scoring mbd: ") + str(scoring_functions.mbd(test_cases,predictions))
139141

140142

141143

0 commit comments

Comments
 (0)