hitrate for nGraphRank

DeepRank · Apr 5, 2019 · 7872208 · 7872208
1 parent 7194eca
commit 7872208
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/iScore/process/plot.py b/iScore/process/plot.py
@@ -100,8 +100,9 @@ def get_hitrate(self,data='iScore',ref='capri',normalize=True):
         index = np.argsort(values)
 
         # revert if necessary
-        if data == 'GraphRank':
-            index = index[::-1]
+        if data.endswith("GraphRank"):
+              index = index[::-1]
+
 
         #sort the data
         values = values[index]