Merge pull request #7 from AGDuran/patch-2

Create FB15k_Tri.py
glorotxa · Mar 17, 2015 · 40a1838 · 40a1838
2 parents afccf9f + a1afe49
commit 40a1838
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/FB15k_Tri.py b/FB15k_Tri.py
@@ -0,0 +1,15 @@
+from FB15k_exp import *
+from FB15k_evaluation import *
+
+savepath='FB15k_Tri'
+datapath='data/'
+launch(op='Tri', ndim=50, marge=0.25, rhoL=5, lremb=0.01, nbatches=150, dataset='FB',
+       totepochs=500, test_all=500, neval=1000, savepath=savepath, datapath=datapath)
+
+print "\n##### EVALUATION #####\n"
+
+MR, T10 = RankingEvalFil(datapath=datapath, dataset='FB', op='Tri',
+        loadmodel=savepath+'/best_valid_model.pkl', Nrel=1345, Nsyn=14951)
+
+print "\n##### MEAN RANK: %s #####\n" % (MR)
+print "\n##### HITS@10: %s #####\n" % (T10)