Merge pull request #301 from vishnukaushik/main

Add Rouge score lens

package.json

@@ -76,6 +76,7 @@
         "react-virtualized-auto-sizer": "^1.0.7",
         "react-window": "^1.8.8",
         "recharts": "^2.4.3",
+        "rouge": "^1.0.3",
         "seedrandom": "^3.0.5",
         "shallowequal": "^1.1.0",
         "short-uuid": "^4.2.2",

src/lenses/RougeScoreLens.tsx

@@ -0,0 +1,39 @@
+import { Lens } from '../types';
+import 'twin.macro';
+import rouge from 'rouge';
+import { formatNumber } from '../dataformat';
+
+const RougeScoreLens: Lens = ({ values }) => {
+    const rouge1 = rouge.n(values[0], values[1], 1);
+    const rouge2 = rouge.n(values[0], values[1], 2);
+    return (
+        <div>
+            <div tw="text-sm truncate px-1 py-0.5 flex items-center h-full">
+                Rouge 1: {formatNumber(rouge1)}
+            </div>
+            <div tw="text-sm truncate px-1 py-0.5 flex items-center h-full">
+                Rouge 2: {formatNumber(rouge2)}
+            </div>
+        </div>
+    );
+};
+
+RougeScoreLens.key = 'RougeScoreView';
+RougeScoreLens.dataTypes = ['str'];
+RougeScoreLens.defaultHeight = 50;
+RougeScoreLens.minHeight = 50;
+RougeScoreLens.maxHeight = 100;
+RougeScoreLens.multi = true;
+RougeScoreLens.displayName = 'ROUGE Score';
+RougeScoreLens.filterAllowedColumns = (allColumns, selectedColumns) => {
+    if (selectedColumns.length === 2) return [];
+    const selectedKeys = selectedColumns.map((selectedCol) => selectedCol.key);
+    return allColumns.filter(({ type, key }) => {
+        return type.kind === 'str' && !selectedKeys.includes(key);
+    });
+};
+RougeScoreLens.isSatisfied = (columns) => {
+    if (columns.length === 2) return true;
+    return false;
+};
+export default RougeScoreLens;

src/lenses/index.ts

@@ -10,6 +10,7 @@ import SequenceLens from './SequenceLens';
 import SpectrogramLens from './SpectrogramLens';
 import TextLens from './TextLens';
 import VideoLens from './VideoLens';
+import RougeScoreLens from './RougeScoreLens';
 
 export const ALL_LENSES = [
     ArrayLens,
@@ -24,4 +25,5 @@ export const ALL_LENSES = [
     HtmlLens,
     MarkdownLens,
     ScalarLens,
+    RougeScoreLens,
 ];

src/rougeScore.d.ts

@@ -0,0 +1 @@
+declare module 'rouge';