From c2a4e3e64e3193fdc75a072940a8689be3b6ba65 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Thu, 13 Mar 2025 11:00:21 +0100
Subject: [PATCH] Add XSPnDE n=2

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

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 <bayesnet/classifiers/KDB.h>
 #include <bayesnet/classifiers/SPODE.h>
 #include <bayesnet/classifiers/XSPODE.h>
+#include <bayesnet/classifiers/XSPnDE.h>
 #include <bayesnet/classifiers/SPnDE.h>
 #include <bayesnet/classifiers/TANLd.h>
 #include <bayesnet/classifiers/KDBLd.h>
@@ -46,4 +47,4 @@ namespace platform {
         Registrar(const std::string& className, function<bayesnet::BaseClassifier* (void)> 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