model.py 文件源码

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

    try:
        os.mkdir(dirname)
    except:
        pass

    if os.path.isfile(model_filename):
        os.remove(model_filename)
    serializers.save_hdf5(model_filename, model)

    params = {
        "vocab_size_enc": model.vocab_size_enc,
        "vocab_size_dec": model.vocab_size_dec,
        "ndim_embedding": model.ndim_embedding,
        "ndim_h": model.ndim_h,
        "num_layers": model.num_layers,
        "densely_connected": model.densely_connected,
        "pooling": model.pooling,
        "zoneout": model.zoneout,
        "dropout": model.dropout,
        "weightnorm": model.weightnorm,
        "wgain": model.wgain,
        "attention": isinstance(model, AttentiveSeq2SeqModel),
    }
    with open(param_filename, "w") as f:
        json.dump(params, f, indent=4, sort_keys=True, separators=(',', ': '))