add other dataset / regression support

egyptdj · Apr 11, 2023 · 40e2bc6 · 40e2bc6
1 parent 5b2b1ad
commit 40e2bc6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -8,7 +8,8 @@
     argv = util.option.parse()
 
     # run and analyze experiment
-    if not argv.no_train: train(argv)
-    if not argv.no_test: test(argv)
-    if not argv.no_analysis and argv.roi=='schaefer': analyze(argv)
+    if not any([argv.train, argv.test, argv.analyze]): argv.train = argv.test = argv.analyze = True
+    if argv.train: train(argv)
+    if argv.test: test(argv)
+    if argv.analyze: analyze(argv)
     exit(0)