30 lines
797 B
Python
Executable File
30 lines
797 B
Python
Executable File
#!/usr/bin/env python3
|
|
"""Run a model."""
|
|
|
|
import os
|
|
import argparse
|
|
from icenet import MODELS
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument(
|
|
'-d',
|
|
required=True,
|
|
help="Directory containing {train,test}.json."
|
|
)
|
|
parser.add_argument(
|
|
'target_fn',
|
|
help=("Target <module>.<fn> in the icenet package. "
|
|
"The function must take the data directory as argument.")
|
|
)
|
|
args = parser.parse_args()
|
|
if not os.path.isdir(args.d):
|
|
raise Exception("Output directory does not exist")
|
|
|
|
model, fn = args.target_fn.split('.')
|
|
print("Running model %s ..." % model)
|
|
fn = getattr(MODELS[model], fn)
|
|
fn(args.d)
|
|
print("Done.")
|