From c7fe1f03d625e8d09c18e75d22c5d9831f00f9af Mon Sep 17 00:00:00 2001 From: "jeremie.desgagne.bouchard" Date: Fri, 13 Oct 2023 01:38:07 -0400 Subject: [PATCH] fix unexported sigmoid funcion for logistic eval in CUDA extension --- benchmarks/regressor.jl | 2 +- experiments/readme_plots_gpu.jl | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmarks/regressor.jl b/benchmarks/regressor.jl index ecc39d0a..d38804b6 100644 --- 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 index 89dc1f07..18fdcf6c 100644 --- 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)