@@ -151,8 +151,9 @@ def check_data(dataset_name: str):
151
151
152
152
153
153
parser = argparse .ArgumentParser ('Interface for preprocessing datasets' )
154
- parser .add_argument ('--dataset_name' , type = str , choices = ['wikipedia' , 'reddit' , 'mooc' , 'lastfm' , 'myket' , 'enron' , 'SocialEvo' , 'uci' ,
155
- 'Flights' , 'CanParl' , 'USLegis' , 'UNtrade' , 'UNvote' , 'Contacts' ],
154
+ parser .add_argument ('--dataset_name' , type = str ,
155
+ choices = ['wikipedia' , 'reddit' , 'mooc' , 'lastfm' , 'myket' , 'enron' , 'SocialEvo' , 'uci' ,
156
+ 'Flights' , 'CanParl' , 'USLegis' , 'UNtrade' , 'UNvote' , 'Contacts' ],
156
157
help = 'Dataset name' , default = 'wikipedia' )
157
158
parser .add_argument ('--node_feat_dim' , type = int , default = 172 , help = 'Number of node raw features' )
158
159
@@ -162,7 +163,8 @@ def check_data(dataset_name: str):
162
163
if args .dataset_name in ['enron' , 'SocialEvo' , 'uci' ]:
163
164
Path ("../processed_data/{}/" .format (args .dataset_name )).mkdir (parents = True , exist_ok = True )
164
165
copy_tree ("../DG_data/{}/" .format (args .dataset_name ), "../processed_data/{}/" .format (args .dataset_name ))
165
- print (f'the original dataset of { args .dataset_name } is unavailable, directly use the processed dataset by previous works.' )
166
+ print (
167
+ f'the original dataset of { args .dataset_name } is unavailable, directly use the processed dataset by previous works.' )
166
168
else :
167
169
# bipartite dataset
168
170
if args .dataset_name in ['wikipedia' , 'reddit' , 'mooc' , 'lastfm' , 'myket' ]:
@@ -171,5 +173,6 @@ def check_data(dataset_name: str):
171
173
preprocess_data (dataset_name = args .dataset_name , bipartite = False , node_feat_dim = args .node_feat_dim )
172
174
print (f'{ args .dataset_name } is processed successfully.' )
173
175
174
- check_data (args .dataset_name )
176
+ if args .dataset_name not in ['myket' ]:
177
+ check_data (args .dataset_name )
175
178
print (f'{ args .dataset_name } passes the checks successfully.' )
0 commit comments