rmontanana/PyWrap
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Completed with errors

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-11-06 17:14:34 +01:00
parent 418fb1444a
commit ec68e142ac
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/main.cc
View File

@@ -44,19 +44,19 @@ int main(int argc, char* argv[])
{ {
cout << "* Begin." << endl; cout << "* Begin." << endl;
{ {
auto [X, y, features, className, states] = loadDataset("iris", true); auto [X, y, features, className, states] = loadDataset("wine", true);
cout << "X: " << X.sizes() << endl; cout << "X: " << X.sizes() << endl;
cout << "y: " << y.sizes() << endl; cout << "y: " << y.sizes() << endl;
auto clf = pywrap::STree(); auto clf = pywrap::STree();
cout << "STree Version: " << clf.version() << endl; cout << "STree Version: " << clf.version() << endl;
// if (true) { if (true) {
// auto svc = pywrap::SVC(); auto svc = pywrap::SVC();
// svc.fit(X, y, features, className, states); svc.fit(X, y, features, className, states);
// cout << "SVC Score: " << svc.score(X, y) << endl; cout << "SVC Score: " << svc.score(X, y) << endl;
// } }
// cout << "Graph: " << endl << clf.graph() << endl; cout << "Graph: " << endl << clf.graph() << endl;
clf.fit(X, y, features, className, states); clf.fit(X, y, features, className, states);
// cout << "STree Score: " << clf.score(X, y) << endl; cout << "STree Score: " << clf.score(X, y) << endl;
auto prediction = clf.predict(X); auto prediction = clf.predict(X);
cout << "Prediction: " << endl << "{"; cout << "Prediction: " << endl << "{";
for (int i = 0; i < prediction.size(0); ++i) { for (int i = 0; i < prediction.size(0); ++i) {