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 {