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

Refactor constructor

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-07-12 03:07:10 +02:00
parent a60b06e2f2
commit 5793c31bc4
3 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/Metrics.cc
View File

@@ -1,30 +1,23 @@
#include "Metrics.hpp"
using namespace std;
namespace bayesnet {
vector<int> linearize(const vector<vector<int>>& vec_vec)
{
vector<int> vec;
for (const auto& v : vec_vec) {
for (auto d : v) {
vec.push_back(d);
}
}
return vec;
}
Metrics::Metrics(torch::Tensor& samples, vector<string>& features, string& className, int classNumStates)
: samples(samples)
, features(features)
, className(className)
, classNumStates(classNumStates)
{
}
Metrics::Metrics(vector<vector<int>>& vsamples, int m, int n, vector<string>& features, string& className, int classNumStates)
Metrics::Metrics(vector<vector<int>>& vsamples, vector<int>& labels, vector<string>& features, string& className, int classNumStates)
: features(features)
, className(className)
, classNumStates(classNumStates)
{
samples = torch::from_blob(linearize(vsamples).data(), { m, n });
samples = torch::zeros({ static_cast<int64_t>(vsamples[0].size()), static_cast<int64_t>(vsamples.size() + 1) }, torch::kInt64);
for (int i = 0; i < vsamples.size(); ++i) {
samples.index_put_({ "...", i }, torch::tensor(vsamples[i], torch::kInt64));
}
samples.index_put_({ "...", -1 }, torch::tensor(labels, torch::kInt64));
}
vector<pair<string, string>> Metrics::doCombinations(const vector<string>& source)
{

2
src/Metrics.hpp
View File

@@ -17,7 +17,7 @@ namespace bayesnet {
double mutualInformation(torch::Tensor&, torch::Tensor&);
public:
Metrics(torch::Tensor&, vector<string>&, string&, int);
Metrics(vector<vector<int>>&, int, int, vector<string>&, string&, int);
Metrics(vector<vector<int>>&, vector<int>&, vector<string>&, string&, int);
vector<float> conditionalEdgeWeights();
};
}