Remove unneeded output

This commit is contained in:
Ricardo Montañana Gómez 2023-08-16 12:36:38 +02:00
parent 80b20f35b4
commit 918a7b4180
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
2 changed files with 0 additions and 3 deletions

View File

@ -45,7 +45,6 @@ namespace bayesnet {
auto mask = samples.index({ -1, "..." }) == value;
margin[value] = mask.sum().item<double>() / samples.size(1);
}
cout << "Margin: " << margin;
for (auto [first, second] : combinations) {
int index_first = find(features.begin(), features.end(), first) - features.begin();
int index_second = find(features.begin(), features.end(), second) - features.begin();

View File

@ -22,8 +22,6 @@ namespace bayesnet {
auto root = mi[mi.size() - 1].first;
// 2. Compute mutual information between each feature and the class
auto weights_matrix = metrics.conditionalEdge(weights);
cout << "*** Weights matrix ***\n";
cout << weights_matrix << "\n";
// 3. Compute the maximum spanning tree
auto mst = metrics.maximumSpanningTree(features, weights_matrix, root);
// 4. Add edges from the maximum spanning tree to the model