refactor to use in python
This commit is contained in:
@@ -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;
|
||||
}
|
@@ -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;
|
||||
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user