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;
}