fixes

Evovest · Sep 11, 2023 · 26ec498 · 26ec498
1 parent c27b7bf
commit 26ec498
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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