refactor to use in python

This commit is contained in:
2023-07-12 01:05:24 +02:00
parent d1eaab6408
commit a60b06e2f2
4 changed files with 33 additions and 5 deletions

View File

@@ -232,6 +232,11 @@ int main(int argc, char** argv)
unsigned int nthreads = std::thread::hardware_concurrency();
cout << "Computer has " << nthreads << " cores." << endl;
auto metrics = bayesnet::Metrics(network.getSamples(), features, className, network.getClassNumStates());
cout << "conditionalEdgeWeight " << endl << metrics.conditionalEdgeWeight() << endl;
cout << "conditionalEdgeWeight " << endl;
auto conditional = metrics.conditionalEdgeWeights();
cout << conditional << endl;
long m = features.size() + 1;
auto matrix = torch::from_blob(conditional.data(), { m, m });
cout << matrix << endl;
return 0;
}

View File

@@ -167,6 +167,9 @@ int main()
}
// Print the resulting 3x3 tensor
std::cout << tensor_3x3 << std::endl;
vector<int> v = { 1,2,3,4,5 };
torch::Tensor t = torch::tensor(v);
cout << t << endl;