diff --git a/LAMDA_TALENT/configs/default/dummy.json b/LAMDA_TALENT/configs/default/dummy.json new file mode 100644 index 0000000..244a815 --- /dev/null +++ b/LAMDA_TALENT/configs/default/dummy.json @@ -0,0 +1,6 @@ +{ + "dummy": { + "model": {}, + "fit": {} + } +} \ No newline at end of file diff --git a/LAMDA_TALENT/configs/opt_space/LinearRegression.json b/LAMDA_TALENT/configs/opt_space/LinearRegression.json new file mode 100644 index 0000000..2c682ec --- /dev/null +++ b/LAMDA_TALENT/configs/opt_space/LinearRegression.json @@ -0,0 +1,8 @@ +{ + "LinearRegression": { + "model": { + + }, + "fit": {} + } +} \ No newline at end of file diff --git a/LAMDA_TALENT/configs/opt_space/NCM.json b/LAMDA_TALENT/configs/opt_space/NCM.json new file mode 100644 index 0000000..87e60af --- /dev/null +++ b/LAMDA_TALENT/configs/opt_space/NCM.json @@ -0,0 +1,8 @@ +{ + "NCM": { + "model": { + + }, + "fit": {} + } +} \ No newline at end of file diff --git a/LAMDA_TALENT/configs/opt_space/NaiveBayes.json b/LAMDA_TALENT/configs/opt_space/NaiveBayes.json new file mode 100644 index 0000000..249f5bd --- /dev/null +++ b/LAMDA_TALENT/configs/opt_space/NaiveBayes.json @@ -0,0 +1,8 @@ +{ + "NaiveBayes": { + "model": { + + }, + "fit": {} + } +} \ No newline at end of file diff --git a/LAMDA_TALENT/configs/opt_space/dummy.json b/LAMDA_TALENT/configs/opt_space/dummy.json new file mode 100644 index 0000000..07153f0 --- /dev/null +++ b/LAMDA_TALENT/configs/opt_space/dummy.json @@ -0,0 +1,8 @@ +{ + "dummy": { + "model": { + + }, + "fit": {} + } +} \ No newline at end of file diff --git a/LAMDA_TALENT/model/utils.py b/LAMDA_TALENT/model/utils.py index 1d374a9..a55d2ab 100644 --- 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")