Update lib/scholar/metrics/ranking.ex

Co-authored-by: Mateusz Sluszniak <56299341+msluszniak@users.noreply.github.com>
elixir-nx · Apr 7, 2024 · 5af66dc · 5af66dc
1 parent ea54261
commit 5af66dc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/scholar/metrics/ranking.ex b/lib/scholar/metrics/ranking.ex
@@ -106,13 +106,13 @@ defmodule Scholar.Metrics.Ranking do
 
       iex> true_relevance = Nx.tensor([10, 0, 0, 1, 5])
       iex> scores = Nx.tensor([0.1, 0.2, 0.3, 4, 70])
-      iex> Scholar.Metrics.Ranking.ndcg_n(true_relevance, scores, [])
+      iex> Scholar.Metrics.Ranking.ndcg_n(true_relevance, scores)
       #Nx.Tensor<
         f32
         0.6956940293312073
       >
       iex> scores = Nx.tensor([0.05, 1.1, 1.0, 0.5, 0.0])
-      iex> Scholar.Metrics.Ranking.ndcg_n(true_relevance, scores, [])
+      iex> Scholar.Metrics.Ranking.ndcg_n(true_relevance, scores)
       #Nx.Tensor<
         f32
         0.4936802089214325
@@ -129,7 +129,7 @@ defmodule Scholar.Metrics.Ranking do
         1.0
       >
   """
-  defn ndcg_n(y_true, y_score, opts) do
+  defn ndcg_n(y_true, y_score, opts \\ []) do
     dcg_n(y_true, y_score, opts) / dcg_n(y_true, y_true, opts)
   end
 end