diff --git a/src/main.cpp b/src/main.cpp index b6b12267505488459d1af859b5b6c9cb4cbfa1ab..575046b01e7b049ee71d86f438532224acaa4d89 100644 --- a/src/main.cpp +++ b/src/main.cpp @@ -57,7 +57,7 @@ int main(int argc, char const *argv[]) } else if(IP._calc_type.compare("classification") == 0) { - SISSOClassifier sisso(IP._feat_space, IP._prop_unit, IP._prop_train, IP._prop_test, IP._task_sizes_train, IP._task_sizes_test, IP._leave_out_inds, IP._n_dim, IP._n_residuals, IP._n_models_store, IP._fix_intercept); + SISSOClassifier sisso(IP._feat_space, IP._prop_unit, IP._prop_train, IP._prop_test, IP._task_sizes_train, IP._task_sizes_test, IP._leave_out_inds, IP._n_dim, IP._n_residuals, IP._n_models_store); sisso.fit(); if(mpi_setup::comm->rank() == 0) diff --git a/src/python/bindings_docstring_keyed.cpp b/src/python/bindings_docstring_keyed.cpp index a9e57c3540303e345f374e14132db0e41ba94a6d..4c6bf47e3613005b27a1ae9dc9e8eba006539cb5 100644 --- a/src/python/bindings_docstring_keyed.cpp +++ b/src/python/bindings_docstring_keyed.cpp @@ -438,8 +438,8 @@ void sisso::descriptor_identifier::registerSISSORegressor() void sisso::descriptor_identifier::registerSISSOClassifier() { - class_<SISSOClassifier, bases<SISSO_DI>>("SISSOClassifier", init<std::shared_ptr<FeatureSpace>, Unit, np::ndarray, np::ndarray, py::list, py::list, py::list, int, int, int, optional<bool>>()) - .def(init<std::shared_ptr<FeatureSpace>, Unit, py::list, py::list, py::list, py::list, py::list, int, int, int, optional<bool>>()) + class_<SISSOClassifier, bases<SISSO_DI>>("SISSOClassifier", init<std::shared_ptr<FeatureSpace>, Unit, np::ndarray, np::ndarray, py::list, py::list, py::list, int, int, int>()) + .def(init<std::shared_ptr<FeatureSpace>, Unit, py::list, py::list, py::list, py::list, py::list, int, int, int>()) .def("fit", &SISSOClassifier::fit, "@DocString_sisso_class_fit@") .add_property("models", &SISSOClassifier::models_py, "@DocString_sisso_class_models_py@") ; diff --git a/tests/test_classification/test_classification.py b/tests/test_classification/test_classification.py index bee82dca22c4cde426476dd5edf19e8aa2516877..5de1e0aad6074e4a4df723c4601810980a1d55d9 100644 --- a/tests/test_classification/test_classification.py +++ b/tests/test_classification/test_classification.py @@ -76,17 +76,7 @@ def test_sisso_classifier(): feat_space = generate_fs(phi_0, prop, [80], op_set, "classification", 1, 10) sisso = SISSOClassifier( - feat_space, - Unit("m"), - prop, - prop_test, - [80], - [20], - list(range(10)), - 2, - 1, - 1, - False, + feat_space, Unit("m"), prop, prop_test, [80], [20], list(range(10)), 2, 1, 1 ) sisso.fit()