Relintai/pmlpp
1
0
Fork 0
mirror of https://github.com/Relintai/pmlpp.git synced 2024-12-22 15:06:47 +01:00
Code Issues Packages Projects Releases Wiki Activity

Registered MLPPBernoulliNB into the ClassDB.

Browse Source
This commit is contained in:
Relintai 2023-02-12 18:26:19 +01:00
parent e33931aeb5
commit b62df601fb
3 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mlpp/bernoulli_nb/bernoulli_nb.cpp
View File

@ -187,3 +187,6 @@ void MLPPBernoulliNB::evaluate() {
} }
} }
} }
void MLPPBernoulliNB::_bind_methods() {
}

10
mlpp/bernoulli_nb/bernoulli_nb.h
View File

@ -10,10 +10,14 @@
#include "core/math/math_defs.h" #include "core/math/math_defs.h"
#include "core/object/reference.h"
#include <map> #include <map>
#include <vector> #include <vector>
class MLPPBernoulliNB { class MLPPBernoulliNB : public Reference {
GDCLASS(MLPPBernoulliNB, Reference);
public: public:
std::vector<real_t> model_set_test(std::vector<std::vector<real_t>> X); std::vector<real_t> model_set_test(std::vector<std::vector<real_t>> X);
real_t model_test(std::vector<real_t> x); real_t model_test(std::vector<real_t> x);
@ -25,11 +29,13 @@ public:
MLPPBernoulliNB(); MLPPBernoulliNB();
~MLPPBernoulliNB(); ~MLPPBernoulliNB();
private: protected:
void compute_vocab(); void compute_vocab();
void compute_theta(); void compute_theta();
void evaluate(); void evaluate();
static void _bind_methods();
// Model Params // Model Params
real_t _prior_1; real_t _prior_1;
real_t _prior_0; real_t _prior_0;

2
register_types.cpp
View File

@ -64,6 +64,7 @@ SOFTWARE.
#include "mlpp/uni_lin_reg/uni_lin_reg.h" #include "mlpp/uni_lin_reg/uni_lin_reg.h"
#include "mlpp/wgan/wgan.h" #include "mlpp/wgan/wgan.h"
#include "mlpp/c_log_log_reg/c_log_log_reg.h" #include "mlpp/c_log_log_reg/c_log_log_reg.h"
#include "mlpp/bernoulli_nb/bernoulli_nb.h"
#include "test/mlpp_tests.h" #include "test/mlpp_tests.h"
@ -110,6 +111,7 @@ void register_pmlpp_types(ModuleRegistrationLevel p_level) {
ClassDB::register_class<MLPPExpReg>(); ClassDB::register_class<MLPPExpReg>();
ClassDB::register_class<MLPPDualSVC>(); ClassDB::register_class<MLPPDualSVC>();
ClassDB::register_class<MLPPCLogLogReg>(); ClassDB::register_class<MLPPCLogLogReg>();
ClassDB::register_class<MLPPBernoulliNB>();
ClassDB::register_class<MLPPDataESimple>(); ClassDB::register_class<MLPPDataESimple>();
ClassDB::register_class<MLPPDataSimple>(); ClassDB::register_class<MLPPDataSimple>();