Merge pull request Franck-Dernoncourt#141 from Franck-Dernoncourt/out…

…putscore Add output_scores option and fix issue with _clean_param_dtypes. ref Franck-Dernoncourt#101
ninghongbo123 · Mar 30, 2019 · aa48235 · aa48235
2 parents a33d268 + 0c0a458
commit aa48235
Show file tree

Hide file tree

Showing 5 changed files with 125 additions and 89 deletions.
diff --git a/neuroner/__main__.py b/neuroner/__main__.py
@@ -45,6 +45,7 @@ def parse_arguments(arguments=None):
     parser.add_argument('--number_of_gpus', required=False, default=None, help='')
     parser.add_argument('--optimizer', required=False, default=None, help='')
     parser.add_argument('--output_folder', required=False, default=None, help='')
+    parser.add_argument('--output_scores', required=False, default=None, help='')
     parser.add_argument('--patience', required=False, default=None, help='')
     parser.add_argument('--plot_format', required=False, default=None, help='')
     parser.add_argument('--pretrained_model_folder', required=False, default=None, help='')