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

TANNew as a TAN variant working

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-08-04 19:42:18 +02:00
parent c568ba111d
commit 64ac8fb4f2
9 changed files with 50 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/BayesNet/Proposal.h
View File

@@ -9,14 +9,16 @@
namespace bayesnet {
class Proposal {
public:
Proposal(vector<vector<int>>& Xv_, vector<int>& yv_);
virtual ~Proposal() = default;
Proposal(vector<vector<int>>& Xv_, vector<int>& yv_, vector<string>& features_, string& className_);
virtual ~Proposal();
protected:
void localDiscretizationProposal(Network& model, vector<string>& features, string className, map<string, vector<int>>& states);
void fit_local_discretization(vector<string>& features, string className, map<string, vector<int>>& states, torch::Tensor& y);
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
map<string, mdlp::CPPFImdlp*> discretizers;
private:
vector<string>& pFeatures;
string& pClassName;
vector<vector<int>>& Xv; // X discrete nxm vector
vector<int>& yv;
};