def load_model(args, X, Y):
if args.init_model:
with ModelFile(open(args.init_model)) as model_in:
return MLP.load(model_in)
else:
return MLP(X, Y, hidden=args.hidden, depth=args.depth, input=len(X), output=len(Y))
def load_model(args):
with ModelFile(open(args.init_model)) as model_in:
return MLP.load(model_in)
