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

Begin implementing KDB

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-07-13 03:15:42 +02:00
parent c5386d66fc
commit 8b0aa5ccfb
8 changed files with 201 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/Metrics.hpp
View File

@@ -14,8 +14,8 @@ namespace bayesnet {
vector<pair<string, string>> doCombinations(const vector<string>&);
double entropy(torch::Tensor&);
double conditionalEntropy(torch::Tensor&, torch::Tensor&);
double mutualInformation(torch::Tensor&, torch::Tensor&);
public:
double mutualInformation(torch::Tensor&, torch::Tensor&);
Metrics(torch::Tensor&, vector<string>&, string&, int);
Metrics(const vector<vector<int>>&, const vector<int>&, const vector<string>&, const string&, const int);
vector<float> conditionalEdgeWeights();