Brute force hhyperparameters

2026-09-25 17:24:13 +02:00
parent c5f95c3c1e
commit ecf749a7d8
2 changed files with 45 additions and 18 deletions
--- a/src/training/decision_tree/model.py
+++ b/src/training/decision_tree/model.py
@@ -64,6 +64,8 @@ class Model:
        print("Test:")
        print("MAE:", test_mae)
        print("MSE:", test_mse)
+        
+        return [train_mae, train_mse, test_mae, test_mse]


    def save_model(self, path):