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

Add states as result in Proposal methods

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-08-12 16:16:17 +02:00
parent 405887f833
commit 182b52a887
5 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/BayesNet/Proposal.h
View File

@@ -14,8 +14,8 @@ namespace bayesnet {
virtual ~Proposal();
protected:
torch::Tensor prepareX(torch::Tensor& X);
void localDiscretizationProposal(map<string, vector<int>>& states, Network& model);
map<string, vector<int>> fit_local_discretization(torch::Tensor& y);
map<string, vector<int>> localDiscretizationProposal(const map<string, vector<int>>& states, Network& model);
map<string, vector<int>> fit_local_discretization(const torch::Tensor& y);
torch::Tensor Xf; // X continuous nxm tensor
torch::Tensor y; // y discrete nx1 tensor
map<string, mdlp::CPPFImdlp*> discretizers;