mirror of
https://github.com/ltcptgeneral/IdealRMT-DecisionTrees.git
synced 2025-09-05 23:07:24 +00:00
cleanup code
This commit is contained in:
@@ -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": [
|
||||
|
Reference in New Issue
Block a user