Add be_hyperparams hyperparameter to ODTE

2024-08-02 12:56:23 +02:00
parent 4defb279b0
commit 7fbdcc1c3d
2 changed files with 20 additions and 1 deletions
--- a/pyclfs/ODTE.cc
+++ b/pyclfs/ODTE.cc
@@ -3,7 +3,7 @@
 namespace pywrap {
    ODTE::ODTE() : PyClassifier("odte", "Odte")
    {
-        validHyperparameters = { "n_jobs", "n_estimators", "random_state" };
+        validHyperparameters = { "n_jobs", "n_estimators", "random_state", "be_hyperparams" };
    }
    int ODTE::getNumberOfNodes() const
    {