model.py 文件源码

def load_model(dirname):
    model_filename = dirname + "/model.hdf5"
    param_filename = dirname + "/params.json"

    if os.path.isfile(param_filename):
        print("loading {} ...".format(param_filename))
        with open(param_filename, "r") as f:
            try:
                params = json.load(f)
            except Exception as e:
                raise Exception("could not load {}".format(param_filename))

        qrnn = RNNModel(**params)

        if os.path.isfile(model_filename):
            print("loading {} ...".format(model_filename))
            serializers.load_hdf5(model_filename, qrnn)

        return qrnn
    else:
        return None