@@ -11,8 +11,8 @@ def main():
11
11
descr = "Train an ember model from a directory with raw feature files"
12
12
parser = argparse .ArgumentParser (prog = prog , description = descr )
13
13
parser .add_argument ("-v" , "--featureversion" , type = int , default = 2 , help = "EMBER feature version" )
14
- parser .add_argument ("-m" , "--metadata" , action = "store_true" , help = "EMBER feature version " )
15
- parser .add_argument ("-t" , "--train" , action = "store_true" , help = "EMBER feature version " )
14
+ parser .add_argument ("-m" , "--metadata" , action = "store_true" , help = "Create metadata CSVs " )
15
+ parser .add_argument ("-t" , "--train" , action = "store_true" , help = "Train an EMBER model " )
16
16
parser .add_argument ("datadir" , metavar = "DATADIR" , type = str , help = "Directory with raw features" )
17
17
parser .add_argument ("--optimize" , help = "gridsearch to find best parameters" , action = "store_true" )
18
18
args = parser .parse_args ()
@@ -25,6 +25,7 @@ def main():
25
25
if not (os .path .exists (X_train_path ) and os .path .exists (y_train_path )):
26
26
print ("Creating vectorized features" )
27
27
ember .create_vectorized_features (args .datadir , args .featureversion )
28
+
28
29
if args .metadata :
29
30
ember .create_metadata (args .datadir )
30
31
0 commit comments