From 26ec4984da34622db414443de1b99f2ef0f899a0 Mon Sep 17 00:00:00 2001 From: "jeremie.desgagne.bouchard" Date: Sun, 10 Sep 2023 20:29:16 -0400 Subject: [PATCH] fixes --- benchmarks/Yahoo-LTRC.jl | 2 +- docs/src/tutorials/ranking-LTRC.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmarks/Yahoo-LTRC.jl b/benchmarks/Yahoo-LTRC.jl index aa6414aa..736caca2 100644 --- a/benchmarks/Yahoo-LTRC.jl +++ b/benchmarks/Yahoo-LTRC.jl @@ -97,7 +97,7 @@ config = EvoTreeRegressor( p_test = m_mse(x_test); test_df = DataFrame(p=p_test, y=y_test, q=q_test) test_df_agg = combine(groupby(test_df, "q"), ["p", "y"] => ndcg => "ndcg") -ndcg_test = round(mean(test_df_agg.ndcg), digits=5) +ndcg_test = round(mean(test_df_agg.ndcg), sigdigits=5) @info "ndcg_test MSE" ndcg_test ##################################### diff --git a/docs/src/tutorials/ranking-LTRC.md b/docs/src/tutorials/ranking-LTRC.md index c63fc7f3..50804932 100644 --- a/docs/src/tutorials/ranking-LTRC.md +++ b/docs/src/tutorials/ranking-LTRC.md @@ -170,7 +170,7 @@ y_test = dtest[:y] p_test = m_logloss(x_test); test_df = DataFrame(p=p_test, y=y_test, q=q_test) test_df_agg = combine(groupby(test_df, "q"), ["p", "y"] => ndcg => "ndcg") -ndcg_test = mean(test_df_agg.ndcg) +ndcg_test = round(mean(test_df_agg.ndcg), sigdigits=5) @info "ndcg_test LogLoss" ndcg_test ┌ Info: ndcg_test LogLoss