Experiment working with smoothing and disc-algo

2024-06-11 13:52:26 +02:00
parent 0f06f8971e
commit e628d80f4c
2 changed files with 4 additions and 4 deletions
--- a/src/main/Experiment.cpp
+++ b/src/main/Experiment.cpp
@@ -194,8 +194,7 @@ namespace platform {
                //
                // Train model
                //
-                clf->setSmoothing(smooth_type);
-                clf->fit(X_train, y_train, features, className, states);
+                clf->fit(X_train, y_train, features, className, states, smooth_type);
                if (!quiet)
                    showProgress(nfold + 1, getColor(clf->getStatus()), "b");
                auto clf_notes = clf->getNotes();