Fix classical methods

qile2000 · Oct 3, 2024 · b1a6401 · b1a6401
1 parent e2b4c1a
commit b1a6401
Show file tree

Hide file tree

Showing 6 changed files with 44 additions and 0 deletions.
diff --git a/LAMDA_TALENT/configs/default/dummy.json b/LAMDA_TALENT/configs/default/dummy.json
@@ -0,0 +1,6 @@
+{
+    "dummy": {
+        "model": {},
+        "fit": {}
+    }
+}
diff --git a/LAMDA_TALENT/configs/opt_space/LinearRegression.json b/LAMDA_TALENT/configs/opt_space/LinearRegression.json
@@ -0,0 +1,8 @@
+{
+    "LinearRegression": {
+        "model": {
+
+        },
+        "fit": {}
+    }
+}
diff --git a/LAMDA_TALENT/configs/opt_space/NCM.json b/LAMDA_TALENT/configs/opt_space/NCM.json
@@ -0,0 +1,8 @@
+{
+    "NCM": {
+        "model": {
+
+        },
+        "fit": {}
+    }
+}
diff --git a/LAMDA_TALENT/configs/opt_space/NaiveBayes.json b/LAMDA_TALENT/configs/opt_space/NaiveBayes.json
@@ -0,0 +1,8 @@
+{
+    "NaiveBayes": {
+        "model": {
+
+        },
+        "fit": {}
+    }
+}
diff --git a/LAMDA_TALENT/configs/opt_space/dummy.json b/LAMDA_TALENT/configs/opt_space/dummy.json
@@ -0,0 +1,8 @@
+{
+    "dummy": {
+        "model": {
+
+        },
+        "fit": {}
+    }
+}
diff --git a/LAMDA_TALENT/model/utils.py b/LAMDA_TALENT/model/utils.py
@@ -818,5 +818,11 @@ def get_method(model):
     elif model == 'svm':
         from model.classical_methods.svm import SvmMethod
         return SvmMethod
+    elif model == 'dummy':
+        from model.classical_methods.dummy import DummyMethod
+        return DummyMethod
+    elif model == 'LinearRegression':
+        from model.classical_methods.lr import LinearRegressionMethod
+        return LinearRegressionMethod
     else:
         raise NotImplementedError("Model \"" + model + "\" not yet implemented")