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)