moreo
/
kernel_authorship
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

added victorian

This commit is contained in:
Alejandro Moreo Fernandez 2020-05-09 11:14:20 +02:00
parent c54032f6d7
commit b22060fbdc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.py
View File

@ -161,7 +161,7 @@ if __name__ == '__main__':
requiredNamed.add_argument('-d', '--dataset', help='Name of the dataset', required=True, type=str) requiredNamed.add_argument('-d', '--dataset', help='Name of the dataset', required=True, type=str)
opt = parser.parse_args() opt = parser.parse_args()
assert opt.dataset in ['enron', 'imdb62', 'blogs'], 'unknown dataset' assert opt.dataset in ['enron', 'imdb62', 'blogs', 'victorian'], 'unknown dataset'
create_path_if_not_exists(opt.output) create_path_if_not_exists(opt.output)
create_path_if_not_exists(opt.log) create_path_if_not_exists(opt.log)