rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

const functions

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-08-08 01:53:41 +02:00
parent ef1bffcac3
commit 323444b74a
27 changed files with 109 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/BayesNet/SPODELd.cc
View File

@@ -2,10 +2,11 @@
namespace bayesnet {
using namespace std;
SPODELd::SPODELd(int root) : SPODE(root), Proposal(dataset, features, className) {}
SPODELd::SPODELd(int root) : SPODE(root), Proposal(dataset, features, className) { cout << "SPODELd constructor" << endl; }
SPODELd& SPODELd::fit(torch::Tensor& X_, torch::Tensor& y_, vector<string>& features_, string className_, map<string, vector<int>>& states_)
{
// This first part should go in a Classifier method called fit_local_discretization o fit_float...
cout << "YOOOOOOOOOOOOOOOOOOOo" << endl;
features = features_;
className = className_;
Xf = X_;
@@ -16,7 +17,6 @@ namespace bayesnet {
// 1st we need to fit the model to build the normal SPODE structure, SPODE::fit initializes the base Bayesian network
SPODE::fit(dataset, features, className, states);
localDiscretizationProposal(states, model);
//model.fit(SPODE::Xv, SPODE::yv, features, className);
return *this;
}
Tensor SPODELd::predict(Tensor& X)
@@ -24,7 +24,11 @@ namespace bayesnet {
auto Xt = prepareX(X);
return SPODE::predict(Xt);
}
vector<string> SPODELd::graph(const string& name)
void SPODELd::test()
{
cout << "SPODELd test" << endl;
}
vector<string> SPODELd::graph(const string& name) const
{
return SPODE::graph(name);
}