From 6bc5058d137daa28184c0494f9380b7832770c59 Mon Sep 17 00:00:00 2001
From: Ines Montani <ines@ines.io>
Date: Thu, 24 Sep 2020 14:53:34 +0200
Subject: [PATCH] Update models directory [ci skip]

---
 website/src/templates/models.js | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/website/src/templates/models.js b/website/src/templates/models.js
index 413f23dc5..cdfe2e46d 100644
--- a/website/src/templates/models.js
+++ b/website/src/templates/models.js
@@ -32,11 +32,17 @@ const MODEL_META = {
     las: 'Labelled dependencies',
     token_acc: 'Tokenization',
     tok: 'Tokenization',
+    lemma: 'Statistical lemmatization',
+    morph: 'Morphological analysis',
     tags_acc: 'Part-of-speech tags (fine grained tags, Token.tag)',
     tag: 'Part-of-speech tags (fine grained tags, Token.tag)',
+    pos: 'Part-of-speech tags (coarse grained tags, Token.pos)',
     ents_f: 'Named entities (F-score)',
     ents_p: 'Named entities (precision)',
     ents_r: 'Named entities (recall)',
+    ner_f: 'Named entities (F-score)',
+    ner_p: 'Named entities (precision)',
+    ner_r: 'Named entities (recall)',
     sent_f: 'Sentence segmentation (F-score)',
     sent_p: 'Sentence segmentation (precision)',
     sent_r: 'Sentence segmentation (recall)',
@@ -88,11 +94,12 @@ function formatVectors(data) {
 }
 
 function formatAccuracy(data) {
+    const exclude = ['speed']
     if (!data) return []
     return Object.keys(data)
         .map(label => {
             const value = data[label]
-            return isNaN(value)
+            return isNaN(value) || exclude.includes(label)
                 ? null
                 : {
                       label,
@@ -109,6 +116,7 @@ function formatModelMeta(data) {
         version: data.version,
         sizeFull: data.size,
         pipeline: data.pipeline,
+        components: data.components,
         notes: data.notes,
         description: data.description,
         sources: data.sources,
@@ -117,7 +125,8 @@ function formatModelMeta(data) {
         license: data.license,
         labels: isEmptyObj(data.labels) ? null : data.labels,
         vectors: formatVectors(data.vectors),
-        accuracy: formatAccuracy(data.performance),
+        // TODO: remove accuracy fallback
+        accuracy: formatAccuracy(data.accuracy || data.performance),
     }
 }