From 3882ebd6e403120bacfe4f7279a43b144fc6d0ba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Sun, 5 May 2024 19:53:14 +0200
Subject: [PATCH] Add SPnDE & A2DE models

---
 src/main/Models.h        | 2 ++
 src/main/modelRegister.h | 4 ++++
 2 files changed, 6 insertions(+)

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 <map>
 #include <bayesnet/BaseClassifier.h>
 #include <bayesnet/ensembles/AODE.h>
+#include <bayesnet/ensembles/A2DE.h>
 #include <bayesnet/ensembles/AODELd.h>
 #include <bayesnet/ensembles/BoostAODE.h>
 #include <bayesnet/classifiers/TAN.h>
 #include <bayesnet/classifiers/KDB.h>
 #include <bayesnet/classifiers/SPODE.h>
+#include <bayesnet/classifiers/SPnDE.h>
 #include <bayesnet/classifiers/TANLd.h>
 #include <bayesnet/classifiers/KDBLd.h>
 #include <bayesnet/classifiers/SPODELd.h>
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",