rmontanana/Pyclassifiers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add Odte hyperparameters

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-08-07 10:52:31 +02:00
parent a3e00d2485
commit 58be7db993
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pyclfs/ODTE.cc
View File

@@ -3,7 +3,7 @@
namespace pywrap { namespace pywrap {
ODTE::ODTE() : PyClassifier("odte", "Odte") 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 int ODTE::getNumberOfNodes() const
{ {