Skip to content
Permalink
Browse files

fixed variable name; thanks Keita!

  • Loading branch information...
srijankr
srijankr committed Jul 19, 2019
1 parent 2056d0a commit 51f06b9f4c47c9319d5f5e22e555bfdde7b88572
Showing with 2 additions and 2 deletions.
  1. +2 −2 evaluate_interaction_prediction.py
@@ -206,12 +206,12 @@
performance_dict = dict()
ranks = validation_ranks
mrr = np.mean([1.0 / r for r in ranks])
rec = sum(np.array(ranks) <= 10)*1.0 / len(ranks)
rec10 = sum(np.array(ranks) <= 10)*1.0 / len(ranks)
performance_dict['validation'] = [mrr, rec10]

ranks = test_ranks
mrr = np.mean([1.0 / r for r in ranks])
rec = sum(np.array(ranks) <= 10)*1.0 / len(ranks)
rec10 = sum(np.array(ranks) <= 10)*1.0 / len(ranks)
performance_dict['test'] = [mrr, rec10]

# PRINT AND SAVE THE PERFORMANCE METRICS

0 comments on commit 51f06b9

Please sign in to comment.
You can’t perform that action at this time.