Skip to content

Commit 75726c8

Browse files
author
Phil Roth
authored
Fixing argument help text
1 parent 565f508 commit 75726c8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

scripts/init_ember.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@ def main():
1111
descr = "Train an ember model from a directory with raw feature files"
1212
parser = argparse.ArgumentParser(prog=prog, description=descr)
1313
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")
1616
parser.add_argument("datadir", metavar="DATADIR", type=str, help="Directory with raw features")
1717
parser.add_argument("--optimize", help="gridsearch to find best parameters", action="store_true")
1818
args = parser.parse_args()
@@ -25,6 +25,7 @@ def main():
2525
if not (os.path.exists(X_train_path) and os.path.exists(y_train_path)):
2626
print("Creating vectorized features")
2727
ember.create_vectorized_features(args.datadir, args.featureversion)
28+
2829
if args.metadata:
2930
ember.create_metadata(args.datadir)
3031

0 commit comments

Comments
 (0)