drmm_tks_quoraqp.config

{
  "net_name": "DRMM_TKS",
  "global":{
      "model_type": "PY",
      "weights_file": "examples/QuoraQP/weights/drmm_tks.quoraqp.weights",
      "save_weights_iters": 500,
      "num_iters": 500,
      "display_interval": 10,
      "test_weights_iters": 500,
      "optimizer": "adam",
      "learning_rate": 0.001
  },
  "inputs": {
    "share": {
        "text1_corpus": "./data/QuoraQP/corpus_preprocessed.txt",
        "text2_corpus": "./data/QuoraQP/corpus_preprocessed.txt",
        "use_dpool": false,
        "embed_size": 300,
        "embed_path": "./data/QuoraQP/embed_glove_d300",
        "vocab_size": 28159,
        "train_embed": false,
        "target_mode": "classification",
        "class_num": 2,
        "text1_maxlen": 20,
        "text2_maxlen": 20
    },
    "train": {
        "input_type": "PointGenerator", 
        "phase": "TRAIN",
        "use_iter": false,
        "query_per_iter": 50,
        "batch_per_iter": 5,
        "batch_size": 1024,
        "relation_file": "./data/QuoraQP/relation_train.txt"
    },
    "valid": {
        "input_type": "PointGenerator", 
        "phase": "EVAL",
        "batch_size": 1024,
        "relation_file": "./data/QuoraQP/relation_valid.txt"
    },
    "test": {
        "input_type": "PointGenerator", 
        "phase": "EVAL",
        "batch_size": 1024,
        "relation_file": "./data/QuoraQP/relation_test.txt"
    },
    "predict": {
        "input_type": "PointGenerator", 
        "phase": "PREDICT",
        "batch_size": 1024,
        "relation_file": "./data/QuoraQP/relation_test.txt"
    }
  },
  "outputs": {
    "predict": {
      "save_format": "TREC",
      "save_path": "predict.test.quoraqp.txt"
    }
  },
  "model": {
    "model_path": "./matchzoo/models/",
    "model_py": "drmm_tks.DRMM_TKS",
    "setting": {
        "topk": 20,
        "num_layers": 3,
        "hidden_sizes": [100, 80, 1] 
    }
  },
  "losses": [ 
    {
       "object_name": "categorical_crossentropy",
       "object_params": {}
    }
  ],
  "metrics": [ "accuracy" ]
}