From b22060fbdcd20f3f802bd59d9d8ca74c964a4123 Mon Sep 17 00:00:00 2001 From: Alex Moreo Date: Sat, 9 May 2020 11:14:20 +0200 Subject: [PATCH] added victorian --- src/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/main.py b/src/main.py index 7edccf7..2dd9758 100644 --- a/src/main.py +++ b/src/main.py @@ -161,7 +161,7 @@ if __name__ == '__main__': requiredNamed.add_argument('-d', '--dataset', help='Name of the dataset', required=True, type=str) 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.log)