From 73c285532706d470429dc2c5517b428269e3b0b3 Mon Sep 17 00:00:00 2001
From: alejandroaldas <52395149+alejandroaldas@users.noreply.github.com>
Date: Tue, 3 Jun 2025 15:03:27 -0500
Subject: [PATCH] Fix MSE call in decision tree

---
 machine_learning/decision_tree.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/machine_learning/decision_tree.py b/machine_learning/decision_tree.py
index 72970431c3fc..008fae48c8af 100644
--- a/machine_learning/decision_tree.py
+++ b/machine_learning/decision_tree.py
@@ -96,7 +96,7 @@ def train(self, x, y):
             return
 
         best_split = 0
-        min_error = self.mean_squared_error(x, np.mean(y)) * 2
+        min_error = self.mean_squared_error(y, np.mean(y)) * 2
 
         """
         loop over all possible splits for the decision tree. find the best split.