add 'l1_flag' in the parser.py for CFKG

xiangwang1223 · Aug 5, 2020 · c03737b · c03737b
1 parent 530327a
commit c03737b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Model/utility/parser.py b/Model/utility/parser.py
@@ -74,5 +74,8 @@ def parse_args():
                         help='whether using attention mechanism')
     parser.add_argument('--use_kge', type=bool, default=True,
                         help='whether using knowledge graph embedding')
+
+    parser.add_argument('--l1_flag', type=bool, default=True,
+                        help='Flase: using the L2 norm, True: using the L1 norm.')
 
-    return parser.parse_args()
+    return parser.parse_args()