Refactor New classifiers to extract predict

This commit is contained in:
2023-08-05 18:39:48 +02:00
parent 1a09ccca4c
commit 7f45495837
6 changed files with 35 additions and 50 deletions

View File

@@ -146,11 +146,6 @@ namespace platform {
auto y_test = y.index({ test_t });
cout << nfold + 1 << ", " << flush;
clf->fit(X_train, y_train, features, className, states);
cout << endl;
auto lines = clf->show();
for (auto line : lines) {
cout << line << endl;
}
nodes[item] = clf->getNumberOfNodes();
edges[item] = clf->getNumberOfEdges();
num_states[item] = clf->getNumberOfStates();