From c03737be46ac26a0b5431efe149828e982ffbbfb Mon Sep 17 00:00:00 2001 From: Xiang Wang Date: Wed, 5 Aug 2020 10:30:32 +0800 Subject: [PATCH] add 'l1_flag' in the parser.py for CFKG --- Model/utility/parser.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/Model/utility/parser.py b/Model/utility/parser.py index 02adc6e..1e588a7 100644 --- 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() \ No newline at end of file + return parser.parse_args()