Minor tweaks

README.md

@@ -1,8 +1,8 @@
 ---
 title: nDCG
 emoji: 👁
-colorFrom: yellow
-colorTo: red
+colorFrom: red
+colorTo: blue
 sdk: gradio
 sdk_version: 3.9.1
 app_file: app.py

app.py

@@ -9,7 +9,7 @@ from evaluate.utils.logging import get_logger
 logger = get_logger(__name__)
 REGEX_YAML_BLOCK = re.compile(r"---[\n\r]+([\S\s]*?)[\n\r]+---[\n\r]")
 
-module = evaluate.load("nDCG.py")
+module = evaluate.load("JP-SystemsX/nDCG")
 
 def launch_gradio_widget(metric):
     """Launches `metric` widget with Gradio."""

nDCG.py

@@ -66,13 +66,13 @@ Examples:
         >>> print(results)
         {'nDCG@3': 0.4123818817534531}
     Example 3-There is only one relevant label, but there is a tie and the model can't decide which one is the one.
-        >>> accuracy_metric = evaluate.load("accuracy")
+        >>> nDCG_metric = evaluate.load("JP-SystemsX/nDCG")
         >>> results = nDCG_metric.compute(references=[[1, 0, 0, 0, 0]], predictions=[[1, 1, 0, 0, 0]], k=1)
         >>> print(results)
         {'nDCG@1': 0.5}
         >>> #That is it calculates both and returns the average of both
     Example 4-The Same as 3, except ignore_ties is set to True.
-        >>> accuracy_metric = evaluate.load("accuracy")
+        >>> nDCG_metric = evaluate.load("JP-SystemsX/nDCG")
         >>> results = nDCG_metric.compute(references=[[1, 0, 0, 0, 0]], predictions=[[1, 1, 0, 0, 0]], k=1, ignore_ties=True)
         >>> print(results)
         {'nDCG@1': 0.0}