diff --git a/DecisionTree.ipynb b/DecisionTree.ipynb index 16d5a4d..11199e1 100644 --- a/DecisionTree.ipynb +++ b/DecisionTree.ipynb @@ -21,7 +21,7 @@ }, { "cell_type": "code", - "execution_count": 32, + "execution_count": null, "id": "d336971a", "metadata": {}, "outputs": [], @@ -61,8 +61,7 @@ " threshold = tree.tree_.threshold\n", " features = [feature_names[i] for i in tree.tree_.feature]\n", " value = tree.tree_.value\n", - " le = '<='\n", - " g = '>'\n", + "\n", " # get ids of child nodes\n", " idx = np.argwhere(left == -1)[:, 0]\n", " \n", @@ -105,7 +104,7 @@ }, { "cell_type": "code", - "execution_count": 33, + "execution_count": null, "id": "b96f3403", "metadata": {}, "outputs": [