alu/IdealRMT-DecisionTrees
1
0
Fork 0
mirror of https://github.com/ltcptgeneral/IdealRMT-DecisionTrees.git synced 2025-09-04 14:27:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

restore TreeCompress and TreeToRMT from main

Browse Source
This commit is contained in:
Arthur Lu
2025-06-11 18:28:18 +00:00
parent e811171a73
commit 7bee40ecf9
3 changed files with 22 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
.DS_Store vendored
View File

Binary file not shown.

37
TreeCompress.ipynb
View File

@@ -2,7 +2,7 @@
"cells": [ "cells": [
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 9, "execution_count": 1,
"id": "ec310f34", "id": "ec310f34",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -14,7 +14,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 10, "execution_count": 2,
"id": "5b54797e", "id": "5b54797e",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -28,7 +28,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 11, "execution_count": 3,
"id": "a38fdb8a", "id": "a38fdb8a",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -38,14 +38,14 @@
"i = 0\n", "i = 0\n",
"\n", "\n",
"path_ids = set()\n", "path_ids = set()\n",
"path_classes = set()\n", "path_classes = tree[\"classes\"]\n",
"\n", "\n",
"# for each path in the tree\n", "# for each path in the tree\n",
"for path in paths:\n", "for path in paths:\n",
"\t# assign a path id \n", "\t# assign a path id \n",
"\tpath[\"id\"] = i\n", "\tpath[\"id\"] = i\n",
"\tpath_ids.add(i)\n", "\tpath_ids.add(i)\n",
"\tpath_classes.add(path[\"classification\"])\n", "\t#path_classes.add(path[\"classification\"])\n",
"\ti += 1\t\n", "\ti += 1\t\n",
"\t# for each condition\n", "\t# for each condition\n",
"\tconditions = path[\"conditions\"]\n", "\tconditions = path[\"conditions\"]\n",
@@ -60,7 +60,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 12, "execution_count": 4,
"id": "2fd4f738", "id": "2fd4f738",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -83,7 +83,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 13, "execution_count": 5,
"id": "98cde024", "id": "98cde024",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -123,7 +123,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 14, "execution_count": 6,
"id": "b6fbadbf", "id": "b6fbadbf",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -171,16 +171,19 @@
"\t\t# for each bucket which encompases the condition's range, add this path's id to the sets \n", "\t\t# for each bucket which encompases the condition's range, add this path's id to the sets \n",
"\t\ti = 0\n", "\t\ti = 0\n",
"\t\tfor bp in breakpoints[feature_name]:\n", "\t\tfor bp in breakpoints[feature_name]:\n",
"\t\t\tin_range = is_in_range(bp, lower, upper)\n", "\t\t\tif is_in_range(bp, lower, upper):\n",
"\t\t\tif in_range:\n",
"\t\t\t\tbuckets_id[feature_name][i].add(ID)\n", "\t\t\t\tbuckets_id[feature_name][i].add(ID)\n",
"\t\t\t\tbuckets_class[feature_name][i].add(Class)\n", "\t\t\t\tbuckets_class[feature_name][i].add(Class)\n",
"\t\t\ti += 1" "\t\t\ti += 1\n",
"\n",
"\t\tif is_in_range(bp+1, lower, upper):\n",
"\t\t\tbuckets_id[feature_name][i].add(ID)\n",
"\t\t\tbuckets_class[feature_name][i].add(Class)"
] ]
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 15, "execution_count": 7,
"id": "0a767971", "id": "0a767971",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -198,13 +201,13 @@
"\tfor i in range(1, len(buckets_id[feature_name]) - 1):\n", "\tfor i in range(1, len(buckets_id[feature_name]) - 1):\n",
"\t\tlower = breakpoints[feature_name][i-1]\n", "\t\tlower = breakpoints[feature_name][i-1]\n",
"\t\tupper = breakpoints[feature_name][i]\n", "\t\tupper = breakpoints[feature_name][i]\n",
"\t\tmembers = buckets_id[feature_name][i]\n", "\t\tpaths = buckets_id[feature_name][i]\n",
"\t\tclasses = buckets_class[feature_name][i]\n", "\t\tclasses = buckets_class[feature_name][i]\n",
"\t\t#print(f\"{feature_name} = [{lower}, {upper}]: {buckets[feature_name][i]}\")\n", "\t\t#print(f\"{feature_name} = [{lower}, {upper}]: {buckets[feature_name][i]}\")\n",
"\t\tcompressed_layers[feature_name].append({\"min\": lower, \"max\": upper, \"paths\": paths, \"classes\": classes})\n", "\t\tcompressed_layers[feature_name].append({\"min\": lower, \"max\": upper, \"paths\": paths, \"classes\": classes})\n",
"\tlower = breakpoints[feature_name][len(breakpoints[feature_name]) - 1]\n", "\tlower = breakpoints[feature_name][len(breakpoints[feature_name]) - 1]\n",
"\tupper = None\n", "\tupper = None\n",
"\tmembers = buckets_id[feature_name][len(buckets_id[feature_name]) - 1]\n", "\tpaths = buckets_id[feature_name][len(buckets_id[feature_name]) - 1]\n",
"\tclasses = buckets_class[feature_name][len(buckets_class[feature_name]) - 1]\n", "\tclasses = buckets_class[feature_name][len(buckets_class[feature_name]) - 1]\n",
"\t#print(f\"{feature_name} = [{lower}, {upper}]: {members}\")\n", "\t#print(f\"{feature_name} = [{lower}, {upper}]: {members}\")\n",
"\tcompressed_layers[feature_name].append({\"min\": lower, \"max\": upper, \"paths\": paths, \"classes\": classes})\n", "\tcompressed_layers[feature_name].append({\"min\": lower, \"max\": upper, \"paths\": paths, \"classes\": classes})\n",
@@ -219,7 +222,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 16, "execution_count": 8,
"id": "561b0bc1", "id": "561b0bc1",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -238,7 +241,7 @@
], ],
"metadata": { "metadata": {
"kernelspec": { "kernelspec": {
"display_name": "Python 3", "display_name": "switch",
"language": "python", "language": "python",
"name": "python3" "name": "python3"
}, },
@@ -252,7 +255,7 @@
"name": "python", "name": "python",
"nbconvert_exporter": "python", "nbconvert_exporter": "python",
"pygments_lexer": "ipython3", "pygments_lexer": "ipython3",
"version": "3.10.6" "version": "3.12.7"
} }
}, },
"nbformat": 4, "nbformat": 4,

4
TreeToRMT.ipynb
View File

@@ -382,7 +382,7 @@
], ],
"metadata": { "metadata": {
"kernelspec": { "kernelspec": {
"display_name": "Python 3", "display_name": "switch",
"language": "python", "language": "python",
"name": "python3" "name": "python3"
}, },
@@ -396,7 +396,7 @@
"name": "python", "name": "python",
"nbconvert_exporter": "python", "nbconvert_exporter": "python",
"pygments_lexer": "ipython3", "pygments_lexer": "ipython3",
"version": "3.10.6" "version": "3.12.7"
} }
}, },
"nbformat": 4, "nbformat": 4,