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

Add detail to fold progress in main

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2023-09-03 16:33:48 +02:00
parent de62d42b74
commit 05b670dfc0
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/Platform/Experiment.cc
View File

@ -159,17 +159,20 @@ namespace platform {
auto y_train = y.index({ train_t });
auto X_test = X.index({ "...", test_t });
auto y_test = y.index({ test_t });
cout << nfold + 1 << ", " << flush;
cout << nfold + 1 << "(a)" << flush;
// Train model
clf->fit(X_train, y_train, features, className, states);
cout << "\b\bb)" << flush;
nodes[item] = clf->getNumberOfNodes();
edges[item] = clf->getNumberOfEdges();
num_states[item] = clf->getNumberOfStates();
train_time[item] = train_timer.getDuration();
auto accuracy_train_value = clf->score(X_train, y_train);
cout << "\b\bc)" << flush;
// Test model
test_timer.start();
auto accuracy_test_value = clf->score(X_test, y_test);
cout << "\b\b\b, " << flush;
test_time[item] = test_timer.getDuration();
accuracy_train[item] = accuracy_train_value;
accuracy_test[item] = accuracy_test_value;