diff --git a/src/main/Models.h b/src/main/Models.h index 1719c92..4746803 100644 --- a/src/main/Models.h +++ b/src/main/Models.h @@ -3,11 +3,13 @@ #include #include #include +#include #include #include #include #include #include +#include #include #include #include diff --git a/src/main/modelRegister.h b/src/main/modelRegister.h index 68fd1de..1120d94 100644 --- a/src/main/modelRegister.h +++ b/src/main/modelRegister.h @@ -6,6 +6,8 @@ static platform::Registrar registrarTLD("TANLd", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::TANLd();}); static platform::Registrar registrarS("SPODE", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::SPODE(2);}); +static platform::Registrar registrarSn("SPnDE", + [](void) -> bayesnet::BaseClassifier* { return new bayesnet::SPnDE({ 0, 1 });}); static platform::Registrar registrarSLD("SPODELd", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::SPODELd(2);}); static platform::Registrar registrarK("KDB", @@ -14,6 +16,8 @@ static platform::Registrar registrarKLD("KDBLd", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::KDBLd(2);}); static platform::Registrar registrarA("AODE", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::AODE();}); +static platform::Registrar registrarA2("A2DE", + [](void) -> bayesnet::BaseClassifier* { return new bayesnet::A2DE();}); static platform::Registrar registrarALD("AODELd", [](void) -> bayesnet::BaseClassifier* { return new bayesnet::AODELd();}); static platform::Registrar registrarBA("BoostAODE",