fix: fetch measures best values with pytorch keys

measure keys may include suffixes added by pytorch.
jolibrain · Jan 13, 2022 · ebe70b8 · ebe70b8
1 parent 82cf2b0
commit ebe70b8
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/stores/modelRepositories/repository.js b/src/stores/modelRepositories/repository.js
@@ -115,15 +115,18 @@ export default class Repository {
 
     if (
       measure_hist &&
-      measure_hist[`${metricKey}_hist`] &&
-      measure_hist[`${metricKey}_hist`].length > 0
+        typeof Object.keys(measure_hist)
+                     .find(k => k.startsWith(metricKey)) !== 'undefined'
     ) {
 
+      const metricHistKey = Object.keys(measure_hist)
+                                  .find(k => k.startsWith(metricKey))
+
       if(['acc', 'accp', 'meaniou', 'map', 'f1'].includes(metricKey)) {
-        value = Math.max.apply(Math, measure_hist[`${metricKey}_hist`])
+        value = Math.max.apply(Math, measure_hist[metricHistKey])
       } else {
         // train_loss, L1_mean_error, mcll, eucll
-        value = Math.min.apply(Math, measure_hist[`${metricKey}_hist`])
+        value = Math.min.apply(Math, measure_hist[metricHistKey])
       }
     }