[api] Adds suffix to percentile metric name

deepjavalibrary · Feb 29, 2024 · 79e2ab8 · 79e2ab8
1 parent ab00bed
commit 79e2ab8
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/api/src/main/java/ai/djl/metric/Metric.java b/api/src/main/java/ai/djl/metric/Metric.java
@@ -105,6 +105,16 @@ private Metric(
         this.dimensions = dimensions;
     }
 
+    /**
+     * Returns a copy of the metric with a new name.
+     *
+     * @param name the new metric name
+     * @return a copy of the metric
+     */
+    public Metric copyOf(String name) {
+        return new Metric(name, value, unit, timestamp, dimensions);
+    }
+
     /**
      * Returns the name of the {@code Metric}.
      *

diff --git a/api/src/main/java/ai/djl/metric/Metrics.java b/api/src/main/java/ai/djl/metric/Metrics.java
@@ -172,7 +172,8 @@ public Metric percentile(String metricName, int percentile) {
         List<Metric> list = new ArrayList<>(metric);
         list.sort(Comparator.comparingDouble(Metric::getValue));
         int index = metric.size() * percentile / 100;
-        return list.get(index);
+        Metric m = list.get(index);
+        return m.copyOf(m.getMetricName() + "_p" + percentile);
     }
 
     /**

diff --git a/api/src/main/java/ai/djl/metric/Unit.java b/api/src/main/java/ai/djl/metric/Unit.java
@@ -41,6 +41,7 @@ public enum Unit {
     GIGABITS_PER_SECOND("Gigabits/Second"),
     TERABITS_PER_SECOND("Terabits/Second"),
     COUNT_PER_SECOND("Count/Second"),
+    COUNT_PER_ITEM("Count/Item"),
     NONE("None");
 
     private static final ConcurrentHashMap<String, Unit> MAP = new ConcurrentHashMap<>();