diff --git a/.gitignore b/.gitignore index c062b91..42fbee0 100644 --- a/.gitignore +++ b/.gitignore @@ -41,3 +41,4 @@ puml/** *.dot diagrams/html/** diagrams/latex/** +.cache diff --git a/src/main/Models.h b/src/main/Models.h index 488e8d3..e0b44c1 100644 --- a/src/main/Models.h +++ b/src/main/Models.h @@ -12,6 +12,7 @@ #include #include #include +#include #include #include #include @@ -46,4 +47,4 @@ namespace platform { Registrar(const std::string& className, function classFactoryFunction); }; } -#endif \ No newline at end of file +#endif diff --git a/src/main/modelRegister.h b/src/main/modelRegister.h index e739c22..b8e1f95 100644 --- a/src/main/modelRegister.h +++ b/src/main/modelRegister.h @@ -37,9 +37,11 @@ namespace platform { [](void) -> bayesnet::BaseClassifier* { return new pywrap::XGBoost();}); static Registrar registrarXSPODE("XSPODE", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::XSpode(0);}); + static Registrar registrarXSPnDE("XSPnDE", + [](void) -> bayesnet::BaseClassifier* { return new bayesnet::XSpnde(0, 1);}); static Registrar registrarXBAODE("XBAODE", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::XBAODE();}); static Registrar registrarXA1DE("XA1DE", [](void) -> bayesnet::BaseClassifier* { return new XA1DE();}); } -#endif \ No newline at end of file +#endif