25 lines
698 B
Python
25 lines
698 B
Python
from datetime import datetime
|
|
from sys import argv
|
|
import numpy as np
|
|
from argparse import ArgumentParser
|
|
|
|
from config import *
|
|
import model
|
|
import data
|
|
import tard_wrangler
|
|
|
|
parser = ArgumentParser()
|
|
parser.add_argument('--model', type=str, help='Specify the model to use')
|
|
parser.add_argument('file', type=str, help='The file to process')
|
|
args = parser.parse_args()
|
|
|
|
if args.model:
|
|
mymodel = model.load_model(args.model)
|
|
else:
|
|
dataset = data.get_data(DATASET_FILE)
|
|
mymodel = model.make_model(dataset)
|
|
timestamp = datetime.now().strftime("%Y%m%d-%H%M%S")
|
|
mymodel.save(MODEL_DIRECTORY + f"model_-_{timestamp}.keras")
|
|
|
|
print(tard_wrangler.full_predict(args.file, args.file + ".norm", mymodel))
|