Fix sample issue

This commit is contained in:
2025-02-04 18:53:23 +01:00
parent 73a4b3d5e5
commit 505edc79ac

View File

@@ -219,7 +219,7 @@ int main(int argc, char** argv)
auto [Xtrain, ytrain] = extract_indices(train, Xd, y);
auto [Xtest, ytest] = extract_indices(test, Xd, y);
clf->fit(Xtrain, ytrain, features, className, states, smoothing);
std::cout << "Nodes: " << clf->geb_managetNumberOfNodes() << std::endl;
std::cout << "Nodes: " << clf->getNumberOfNodes() << std::endl;
nodes += clf->getNumberOfNodes();
score_train = clf->score(Xtrain, ytrain);
score_test = clf->score(Xtest, ytest);