gru_pairwise.json 503 字节
Newer Older
Y
Yibing Liu 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
{
    "net": {
        "module_name": "gru",
        "class_name": "GRU",
        "emb_dim": 128, 
        "gru_dim": 128,
        "hidden_dim": 128
    },
    "loss": {
        "module_name": "hinge_loss",
        "class_name": "HingeLoss",
        "margin": 0.1
    },
    "optimizer": {
        "class_name": "AdamOptimizer",
        "learning_rate": 0.001,
        "beta1": 0.9,
        "beta2": 0.999,
        "epsilon": 1e-08
    },
    "task_mode": "pairwise",
    "model_path": "gru_pairwise"
}