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

refactor main

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2023-07-15 20:23:43 +02:00
parent e3863387bb
commit 29aca0b35f
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
3 changed files with 24 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
sample/main.cc
View File

@ -259,30 +259,30 @@ int main(int argc, char** argv)
} }
states[className] = vector<int>( states[className] = vector<int>(
maxes[className]); maxes[className]);
cout << "****************** KDB ******************" << endl; // cout << "****************** KDB ******************" << endl;
auto kdb = bayesnet::KDB(2); // auto kdb = bayesnet::KDB(2);
kdb.fit(Xd, y, features, className, states); // kdb.fit(Xd, y, features, className, states);
for (auto line : kdb.show()) { // for (auto line : kdb.show()) {
cout << line << endl; // cout << line << endl;
} // }
cout << "Score: " << kdb.score(Xd, y) << endl; // cout << "Score: " << kdb.score(Xd, y) << endl;
cout << "****************** KDB ******************" << endl; // cout << "****************** KDB ******************" << endl;
cout << "****************** SPODE ******************" << endl; // cout << "****************** SPODE ******************" << endl;
auto spode = bayesnet::SPODE(2); // auto spode = bayesnet::SPODE(2);
spode.fit(Xd, y, features, className, states); // spode.fit(Xd, y, features, className, states);
for (auto line : spode.show()) { // for (auto line : spode.show()) {
cout << line << endl; // cout << line << endl;
} // }
cout << "Score: " << spode.score(Xd, y) << endl; // cout << "Score: " << spode.score(Xd, y) << endl;
cout << "****************** SPODE ******************" << endl; // cout << "****************** SPODE ******************" << endl;
cout << "****************** AODE ******************" << endl; // cout << "****************** AODE ******************" << endl;
auto aode = bayesnet::AODE(); // auto aode = bayesnet::AODE();
aode.fit(Xd, y, features, className, states); // aode.fit(Xd, y, features, className, states);
for (auto line : aode.show()) { // for (auto line : aode.show()) {
cout << line << endl; // cout << line << endl;
} // }
cout << "Score: " << aode.score(Xd, y) << endl; // cout << "Score: " << aode.score(Xd, y) << endl;
cout << "****************** AODE ******************" << endl; // cout << "****************** AODE ******************" << endl;
cout << "****************** TAN ******************" << endl; cout << "****************** TAN ******************" << endl;
auto tan = bayesnet::TAN(); auto tan = bayesnet::TAN();
tan.fit(Xd, y, features, className, states); tan.fit(Xd, y, features, className, states);

1
src/Metrics.cc
View File

@ -127,6 +127,5 @@ namespace bayesnet {
auto result = vector<pair<int, int>>(); auto result = vector<pair<int, int>>();
auto mst = MST(features, weights, root); auto mst = MST(features, weights, root);
return mst.maximumSpanningTree(); return mst.maximumSpanningTree();
} }
} }

1
src/Mst.cc
View File

@ -108,7 +108,6 @@ namespace bayesnet {
} }
} }
g.kruskal_algorithm(); g.kruskal_algorithm();
//g.display_mst();
auto mst = g.get_mst(); auto mst = g.get_mst();
return reorder(mst, root); return reorder(mst, root);
} }