rmontanana/Platform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Including XA1DE model #5

Merged
rmontanana merged 33 commits from XA1DE into main 2025-03-20 14:58:38 +00:00
Conversation 0 Commits 33 Files Changed 35 +5510 -132
32 changed files with 5510 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit c9ab88e475 - Show all commits

4
src/experimental_clfs/XA1DE.cpp
View File

@@ -14,7 +14,7 @@ namespace platform {
auto y = TensorUtils::to_vector<int>(dataset.index({ -1, "..." }));
int num_instances = X[0].size();
weights_ = torch::full({ num_instances }, 1.0);
normalize_weights(num_instances);
//normalize_weights(num_instances);
aode_.fit(X, y, features, className, states, weights_, true, smoothing);
}
}
}

3
src/main/Models.h
View File

@@ -6,13 +6,14 @@
#include <bayesnet/ensembles/A2DE.h>
#include <bayesnet/ensembles/AODELd.h>
#include <bayesnet/ensembles/XBAODE.h>
#include <bayesnet/ensembles/XBA2DE.h>
#include <bayesnet/ensembles/BoostAODE.h>
#include <bayesnet/ensembles/BoostA2DE.h>
#include <bayesnet/classifiers/TAN.h>
#include <bayesnet/classifiers/KDB.h>
#include <bayesnet/classifiers/SPODE.h>
#include <bayesnet/classifiers/XSPODE.h>
#include <bayesnet/classifiers/XSPnDE.h>
#include <bayesnet/classifiers/XSP2DE.h>
#include <bayesnet/classifiers/SPnDE.h>
#include <bayesnet/classifiers/TANLd.h>
#include <bayesnet/classifiers/KDBLd.h>

6
src/main/modelRegister.h
View File

@@ -37,10 +37,12 @@ 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 registrarXSP2DE("XSP2DE",
[](void) -> bayesnet::BaseClassifier* { return new bayesnet::XSp2de(0, 1);});
static Registrar registrarXBAODE("XBAODE",
[](void) -> bayesnet::BaseClassifier* { return new bayesnet::XBAODE();});
static Registrar registrarXBA2DE("XBA2DE",
[](void) -> bayesnet::BaseClassifier* { return new bayesnet::XBA2DE();});
static Registrar registrarXA1DE("XA1DE",
[](void) -> bayesnet::BaseClassifier* { return new XA1DE();});
}