Add ROUGE score lens to python API

Renumics · Nov 20, 2023 · 25bc544 · 25bc544
1 parent 9e73a80
commit 25bc544
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/renumics/spotlight/layout/lenses.py b/renumics/spotlight/layout/lenses.py
@@ -232,3 +232,12 @@ def spectrogram(
         name=name,
         settings={"freqScale": frequency_scale, "ampScale": amplitude_scale},
     )
+
+
+def rouge_score(column: str, reference_column: str, name: Optional[str] = None) -> Lens:
+    """
+    Add ROUGE score viewer to Spotlight inspector widget.
+
+    Supports a pair of string columns.
+    """
+    return Lens(type="RougeScoreView", columns=[column, reference_column], name=name)