From 58be7db9931b3f506c725a72536629f423948835 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?= Date: Wed, 7 Aug 2024 10:52:31 +0200 Subject: [PATCH] Add Odte hyperparameters --- pyclfs/ODTE.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyclfs/ODTE.cc b/pyclfs/ODTE.cc index 09ea99b..32c0e60 100644 --- 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", "be_hyperparams" }; + validHyperparameters = { "n_jobs", "n_estimators", "random_state", "max_samples", "max_features", "be_hyperparams" }; } int ODTE::getNumberOfNodes() const {