Example #1
0
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))
Example #2
0
def load_model(args):
    with ModelFile(open(args.init_model)) as model_in:
        return MLP.load(model_in)