rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 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
3 changed files with 24 additions and 26 deletions
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>(
maxes[className]);
cout << "****************** KDB ******************" << endl;
auto kdb = bayesnet::KDB(2);
kdb.fit(Xd, y, features, className, states);
for (auto line : kdb.show()) {
cout << line << endl;
}
cout << "Score: " << kdb.score(Xd, y) << endl;
cout << "****************** KDB ******************" << endl;
cout << "****************** SPODE ******************" << endl;
auto spode = bayesnet::SPODE(2);
spode.fit(Xd, y, features, className, states);
for (auto line : spode.show()) {
cout << line << endl;
}
cout << "Score: " << spode.score(Xd, y) << endl;
cout << "****************** SPODE ******************" << endl;
cout << "****************** AODE ******************" << endl;
auto aode = bayesnet::AODE();
aode.fit(Xd, y, features, className, states);
for (auto line : aode.show()) {
cout << line << endl;
}
cout << "Score: " << aode.score(Xd, y) << endl;
cout << "****************** AODE ******************" << endl;
// cout << "****************** KDB ******************" << endl;
// auto kdb = bayesnet::KDB(2);
// kdb.fit(Xd, y, features, className, states);
// for (auto line : kdb.show()) {
// cout << line << endl;
// }
// cout << "Score: " << kdb.score(Xd, y) << endl;
// cout << "****************** KDB ******************" << endl;
// cout << "****************** SPODE ******************" << endl;
// auto spode = bayesnet::SPODE(2);
// spode.fit(Xd, y, features, className, states);
// for (auto line : spode.show()) {
// cout << line << endl;
// }
// cout << "Score: " << spode.score(Xd, y) << endl;
// cout << "****************** SPODE ******************" << endl;
// cout << "****************** AODE ******************" << endl;
// auto aode = bayesnet::AODE();
// aode.fit(Xd, y, features, className, states);
// for (auto line : aode.show()) {
// cout << line << endl;
// }
// cout << "Score: " << aode.score(Xd, y) << endl;
// cout << "****************** AODE ******************" << endl;
cout << "****************** TAN ******************" << endl;
auto tan = bayesnet::TAN();
tan.fit(Xd, y, features, className, states);