File size: 1,194 Bytes
5381b52 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import argparse
from . import train, predict, process, find_matches
def main():
parser = argparse.ArgumentParser(
description="Command-line interface for marcai package"
)
subparsers = parser.add_subparsers(required=True)
train_parser = subparsers.add_parser(
"train", parents=[train.args_parser()], help="Train a model", add_help=False
)
predict_parser = subparsers.add_parser(
"predict",
parents=[predict.args_parser()],
help="Make predictions using a trained model",
add_help=False,
)
process_parser = subparsers.add_parser(
"process", parents=[process.args_parser()], help="Process data", add_help=False
)
find_matches_parser = subparsers.add_parser(
"find_matches",
parents=[find_matches.args_parser()],
help="Find matches in data",
add_help=False,
)
train_parser.set_defaults(func=train.main)
predict_parser.set_defaults(func=predict.main)
process_parser.set_defaults(func=process.main)
find_matches_parser.set_defaults(func=find_matches.main)
args = parser.parse_args()
args.func(args)
if __name__ == "__main__":
main()
|