predict function

PyML-studio · Nov 16, 2024 · e402b4a · e402b4a
1 parent 52fa9d8
commit e402b4a
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/decision_tree/decision_tree.ipynb b/decision_tree/decision_tree.ipynb
@@ -211,7 +211,18 @@
     "    def _most_common_label(self, y):\n",
     "        return Counter(y).most_common(1)[0][0]\n",
     "    \n",
-    "    \n"
+    "    def predict(self, X):\n",
+    "        predictions = [self._predict(inputs) for inputs in X]\n",
+    "        return np.array(predictions)\n",
+    "    \n",
+    "    def _predict_sample(self, x, tree):\n",
+    "        if tree['leaf']:\n",
+    "            return tree['value']\n",
+    "        feature_value = x[tree['feature_index']]\n",
+    "        if feature_value < tree['threshold']:\n",
+    "            return self._predict_sample(x, tree['left'])\n",
+    "        else:\n",
+    "            return self._predict_sample(x, tree['right'])\n"
    ]
   }
  ],