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

Refactor library and models to lighten data stored

Browse Source 
Refactro Ensemble to inherit from Classifier insted of BaseClassifier
This commit is contained in:
Ricardo Montañana Gómez
2023-08-07 12:49:37 +02:00
parent e74565ba01
commit 06db8f51ce
28 changed files with 134 additions and 189 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/BayesNet/Proposal.h
View File

@@ -10,20 +10,21 @@
namespace bayesnet {
class Proposal {
public:
Proposal(vector<vector<int>>& Xv_, vector<int>& yv_, vector<string>& features_, string& className_);
Proposal(torch::Tensor& pDataset, vector<string>& features_, string& className_);
virtual ~Proposal();
protected:
torch::Tensor prepareX(torch::Tensor& X);
void localDiscretizationProposal(map<string, vector<int>>& states, Network& model);
void fit_local_discretization(map<string, vector<int>>& states, torch::Tensor& y);
torch::Tensor Xf; // X continuous nxm tensor
torch::Tensor y; // y discrete nx1 tensor
map<string, mdlp::CPPFImdlp*> discretizers;
int m, n;
private:
torch::Tensor& pDataset; // (n+1)xm tensor
vector<string>& pFeatures;
string& pClassName;
vector<vector<int>>& Xv; // X discrete nxm vector
vector<int>& yv;
};
}
#endif
#endif