Support query vertex diversity of single operators.

hxzd5568 · Nov 7, 2022 · 4547301 · 4547301
1 parent 9f12a4b
commit 4547301
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/gencog/metric/div.py b/gencog/metric/div.py
@@ -91,9 +91,12 @@ def record(self, opr: Operation):
         # Record hash for this operation
         self._hash[op].add(h)
 
+    def __getitem__(self, op: Op):
+        return min(len(self._hash[op]) / self._space[op].total, 1)
+
     @property
     def op_div(self):
-        return np.array([min(len(self._hash[op]) / self._space[op].total, 1) for op in self._ops])
+        return np.array([self[op] for op in self._ops])
 
     @property
     def result(self) -> float: