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

Fix mistake with references

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2023-07-12 03:23:28 +02:00
parent 5793c31bc4
commit c5386d66fc
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/Metrics.cc
View File

@ -8,7 +8,7 @@ namespace bayesnet {
, classNumStates(classNumStates) , classNumStates(classNumStates)
{ {
} }
Metrics::Metrics(vector<vector<int>>& vsamples, vector<int>& labels, vector<string>& features, string& className, int classNumStates) Metrics::Metrics(const vector<vector<int>>& vsamples, const vector<int>& labels, const vector<string>& features, const string& className, const int classNumStates)
: features(features) : features(features)
, className(className) , className(className)
, classNumStates(classNumStates) , classNumStates(classNumStates)

6
src/Metrics.hpp
View File

@ -8,8 +8,8 @@ namespace bayesnet {
class Metrics { class Metrics {
private: private:
torch::Tensor samples; torch::Tensor samples;
vector<string>& features; vector<string> features;
string& className; string className;
int classNumStates; int classNumStates;
vector<pair<string, string>> doCombinations(const vector<string>&); vector<pair<string, string>> doCombinations(const vector<string>&);
double entropy(torch::Tensor&); double entropy(torch::Tensor&);
@ -17,7 +17,7 @@ namespace bayesnet {
double mutualInformation(torch::Tensor&, torch::Tensor&); double mutualInformation(torch::Tensor&, torch::Tensor&);
public: public:
Metrics(torch::Tensor&, vector<string>&, string&, int); Metrics(torch::Tensor&, vector<string>&, string&, int);
Metrics(vector<vector<int>>&, vector<int>&, vector<string>&, string&, int); Metrics(const vector<vector<int>>&, const vector<int>&, const vector<string>&, const string&, const int);
vector<float> conditionalEdgeWeights(); vector<float> conditionalEdgeWeights();
}; };
} }