fix unexported sigmoid funcion for logistic eval in CUDA extension

Evovest · Oct 13, 2023 · c7fe1f0 · c7fe1f0
1 parent ada64b8
commit c7fe1f0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/benchmarks/regressor.jl b/benchmarks/regressor.jl
@@ -23,7 +23,7 @@ nobs = Int(1e6)
 num_feat = Int(100)
 nrounds = 200
 max_depth = 6
-tree_type = "oblivious"
+tree_type = "binary"
 T = Float64
 nthread = Base.Threads.nthreads()
 @info "testing with: $nobs observations | $num_feat features. nthread: $nthread | tree_type : $tree_type | max_depth : $max_depth"

diff --git a/experiments/readme_plots_gpu.jl b/experiments/readme_plots_gpu.jl
@@ -11,7 +11,7 @@ using EvoTrees: predict, sigmoid, logit
 # using ProfileView
 
 # prepare a dataset
-tree_type = "oblivious" # binary/oblivious
+tree_type = "binary" # binary/oblivious
 device = "gpu"
 
 Random.seed!(123)