Add changeModel to b_manage

This commit is contained in:
2025-02-04 17:34:00 +01:00
parent cbe8f4c79c
commit 73a4b3d5e5
4 changed files with 37 additions and 6 deletions

View File

@@ -219,14 +219,14 @@ 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->getNumberOfNodes() << std::endl;
std::cout << "Nodes: " << clf->geb_managetNumberOfNodes() << std::endl;
nodes += clf->getNumberOfNodes();
score_train = clf->score(Xtrain, ytrain);
score_test = clf->score(Xtest, ytest);
}
if (dump_cpt) {
std::cout << "--- CPT Tables ---" << std::endl;
clf->dump_cpt();
std::cout << clf->dump_cpt();
}
total_score_train += score_train;
total_score += score_test;