Bugfix

src/models/train_model.py

@@ -62,7 +62,7 @@ if __name__ == '__main__':
         model_results[name]["Hits@3"] = results.data[('hits_at_3', 'both', 'realistic')]
         model_results[name]["Hits@10"] = results.data[('hits_at_10', 'both', 'realistic')]
         model_results[name]["MR"] = results.data[('arithmetic_mean_rank', 'both', 'realistic')]
-        model_results[name]["MMR"] = results.data[('inverse_harmonic_mean_rank', 'both', 'realistic')]
+        model_results[name]["MRR"] = results.data[('inverse_harmonic_mean_rank', 'both', 'realistic')]
 
     for name in model_names:
         print(name)