🐛 Fix (non-)use of path parameter in Models

Additionally pretty-prints model file for easier editing
mlexchange · Mar 5, 2024 · b51c998 · b51c998
1 parent 132a5ea
commit b51c998
Show file tree

Hide file tree

Showing 3 changed files with 71 additions and 8 deletions.
diff --git a/assets/mode_description.json b/assets/mode_description.json
diff --git a/assets/models.json b/assets/models.json
@@ -0,0 +1,69 @@
+{
+    "contents": [
+        {
+            "model_name": "random_forest",
+            "version": "1.0.0",
+            "type": "supervised",
+            "user": "mlexchange team",
+            "uri": "xxx",
+            "application": [
+                "classification",
+                "segmentation"
+            ],
+            "description": "xxx",
+            "gui_parameters": [
+                {
+                    "type": "int",
+                    "name": "num-tree",
+                    "title": "Number of Trees",
+                    "param_key": "n_estimators",
+                    "value": "30"
+                },
+                {
+                    "type": "int",
+                    "name": "tree-depth",
+                    "title": "Tree Depth",
+                    "param_key": "max_depth",
+                    "value": "8"
+                }
+            ],
+            "cmd": [
+                "xxx"
+            ],
+            "reference": "Adapted from Dash Plotly image segmentation example"
+        },
+        {
+            "model_name": "kmeans",
+            "version": "1.0.0",
+            "type": "unsupervised",
+            "user": "mlexchange team",
+            "uri": "xxx",
+            "application": [
+                "segmentation",
+                "clustering"
+            ],
+            "description": "xxx",
+            "gui_parameters": [
+                {
+                    "type": "int",
+                    "name": "num-cluster",
+                    "title": "Number of Clusters",
+                    "param_key": "n_clusters",
+                    "value": "2"
+                },
+                {
+                    "type": "int",
+                    "name": "num-iter",
+                    "title": "Max Iteration",
+                    "param_key": "max_iter",
+                    "value": "300"
+                }
+            ],
+            "cmd": [
+                "xxx",
+                "xxxx"
+            ],
+            "reference": "Nicholas Schwartz & Howard Yanxon"
+        }
+    ]
+}
diff --git a/utils/content_registry.py b/utils/content_registry.py
@@ -3,9 +3,9 @@
 
 
 class Models:
-    def __init__(self, modelfile_path="./assets/mode_description.json"):
+    def __init__(self, modelfile_path="./assets/models.json"):
         self.path = modelfile_path
-        f = open("./assets/mode_description.json")
+        f = open(self.path)
 
         self.contents = json.load(f)["contents"]
         self.modelname_list = [content["model_name"] for content in self.contents]