Add default value of has_edge_importance

python/dglke/kvclient.py

@@ -48,6 +48,10 @@ def __init__(self):
                           help='IP configuration file of kvstore')
         self.add_argument('--num_client', type=int, default=1,
                           help='Number of client on each machine.')
+        self.add_argument('--has_edge_importance', action='store_true',
+                          help='Allow providing edge importance score for each'\
+                          ' edge during training. The positive score will be '\
+                          'adjusted as pos_score = pos_score * edge_importance')
 
 def get_long_tail_partition(n_relations, n_machine):
     """Relation types has a long tail distribution for many dataset.

python/dglke/kvserver.py

@@ -95,6 +95,10 @@ def __init__(self):
                           help='IP configuration file of kvstore')
         self.add_argument('--total_client', type=int, default=1,
                           help='Total number of client worker nodes')
+        self.add_argument('--has_edge_importance', action='store_true',
+                          help='Allow providing edge importance score for each'\
+                          ' edge during training. The positive score will be '\
+                          'adjusted as pos_score = pos_score * edge_importance')
 
 def get_server_data(args, machine_id):
    """Get data from data_path/dataset/part_machine_id