update default settings

hwwang55 · Oct 29, 2018 · ba7ce72 · ba7ce72
1 parent 05459f1
commit ba7ce72
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main.py b/src/main.py
@@ -7,7 +7,7 @@
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--dataset', type=str, default='movie', help='which dataset to use')
-parser.add_argument('--dim', type=int, default=4, help='dimension of entity and relation embeddings')
+parser.add_argument('--dim', type=int, default=16, help='dimension of entity and relation embeddings')
 parser.add_argument('--n_hop', type=int, default=2, help='maximum hops')
 parser.add_argument('--kge_weight', type=float, default=0.01, help='weight of the KGE term')
 parser.add_argument('--l2_weight', type=float, default=1e-7, help='weight of the l2 regularization term')
@@ -20,12 +20,12 @@
 parser.add_argument('--using_all_hops', type=bool, default=True,
                     help='whether using outputs of all hops or just the last hop when making prediction')
 
-# default settings for Book-Crossing
 '''
+# default settings for Book-Crossing
 parser = argparse.ArgumentParser()
 parser.add_argument('--dataset', type=str, default='book', help='which dataset to use')
 parser.add_argument('--dim', type=int, default=4, help='dimension of entity and relation embeddings')
-parser.add_argument('--n_hop', type=int, default=3, help='maximum hops')
+parser.add_argument('--n_hop', type=int, default=2, help='maximum hops')
 parser.add_argument('--kge_weight', type=float, default=1e-2, help='weight of the KGE term')
 parser.add_argument('--l2_weight', type=float, default=1e-5, help='weight of the l2 regularization term')
 parser.add_argument('--lr', type=float, default=1e-3, help='learning rate')