diff --git a/test/Project.toml b/test/Project.toml index 980db347..e1909c9c 100644 --- a/test/Project.toml +++ b/test/Project.toml @@ -21,7 +21,7 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40" [compat] Distributions = "0.25" DynamicHMC = "3" -EvoTrees = "0.14.7" +EvoTrees = "0.14.7, 0.15" FFTW = "1.1" LogDensityProblems = "0.12, 1, 2" LogExpFunctions = "0.3" diff --git a/test/rstar.jl b/test/rstar.jl index cd445789..848fb221 100644 --- a/test/rstar.jl +++ b/test/rstar.jl @@ -25,8 +25,8 @@ end @testset "samples input type: $wrapper" for wrapper in [Vector, Array, Tables.table] # In practice, probably you want to use EvoTreeClassifier with early stopping classifiers = ( - EvoTreeClassifier(; nrounds=100, eta=0.3), - Pipeline(EvoTreeClassifier(; nrounds=100, eta=0.3); operation=predict_mode), + EvoTreeClassifier(; nrounds=1_000, eta=0.1), + Pipeline(EvoTreeClassifier(; nrounds=1_000, eta=0.1); operation=predict_mode), DecisionTreeClassifier(), SVC(), XGBoostClassifiers..., @@ -128,9 +128,9 @@ end # In practice, probably you want to use EvoTreeClassifier with early stopping rng = MersenneTwister(42) classifiers = ( - EvoTreeClassifier(; rng=rng, nrounds=100, eta=0.3), + EvoTreeClassifier(; rng=rng, nrounds=1_000, eta=0.1), Pipeline( - EvoTreeClassifier(; rng=rng, nrounds=100, eta=0.3); operation=predict_mode + EvoTreeClassifier(; rng=rng, nrounds=1_000, eta=0.1); operation=predict_mode ), DecisionTreeClassifier(; rng=rng), SVC(),