diff --git a/machine_learning/decision_tree.py b/machine_learning/decision_tree.py index 51f600cace41..8617803e88aa 100644 --- a/machine_learning/decision_tree.py +++ b/machine_learning/decision_tree.py @@ -5,6 +5,7 @@ """ import numpy as np +import scoring_functions as scoring_functions class Decision_Tree: def __init__(self, depth = 5, min_leaf_size = 5): @@ -133,6 +134,11 @@ def main(): print("Test values: " + str(test_cases)) print("Predictions: " + str(predictions)) print("Average error: " + str(avg_error)) + print("Scoring mae: ") + str(scoring_functions.mae(test_cases,predictions)) + print("Scoring rmse: ") + str(scoring_functions.rmse(test_cases,predictions)) + print("Scoring rmsle: ") + str(scoring_functions.rmsle(test_cases,predictions)) + print("Scoring mbd: ") + str(scoring_functions.mbd(test_cases,predictions)) + if __name__ == '__main__':