anmm_wikiqa.config

{
  "net_name": "ANMM",
  "global":{
      "model_type": "PY",
      "weights_file": "examples/wikiqa/weights/anmm.wikiqa.weights",
      "save_weights_iters": 10,
      "num_iters": 400,
      "display_interval": 10,
      "test_weights_iters": 400,
      "optimizer": "adadelta",
      "learning_rate": 1.0
  },
  "inputs": {
    "share": {
        "text1_corpus": "./data/WikiQA/corpus_preprocessed.txt",
        "text2_corpus": "./data/WikiQA/corpus_preprocessed.txt",
        "use_dpool": false,
        "embed_size": 300,
        "embed_path": "./data/WikiQA/embed.idf",
        "vocab_size": 18670,
        "train_embed": false,
        "target_mode": "ranking",
        "bin_num": 20,
        "text1_maxlen": 10,
        "text2_maxlen": 40
    },
    "train": {
        "input_type": "DRMM_PairGenerator",
        "phase": "TRAIN",
        "use_iter": false,
        "query_per_iter": 50,
        "batch_per_iter": 5,
        "batch_size": 100,
        "relation_file": "./data/WikiQA/relation_train.txt",
        "hist_feats_file": "./data/WikiQA/relation_train.binsum-20.txt"
    },
    "valid": {
        "input_type": "DRMM_ListGenerator",
        "phase": "EVAL",
        "batch_list": 10,
        "relation_file": "./data/WikiQA/relation_valid.txt",
        "hist_feats_file": "./data/WikiQA/relation_valid.binsum-20.txt"
    },
    "test": {
        "input_type": "DRMM_ListGenerator",
        "phase": "EVAL",
        "batch_list": 10,
        "relation_file": "./data/WikiQA/relation_test.txt",
        "hist_feats_file": "./data/WikiQA/relation_test.binsum-20.txt"
    },
    "predict": {
        "input_type": "DRMM_ListGenerator",
        "phase": "PREDICT",
        "batch_list": 10,
        "relation_file": "./data/WikiQA/relation_test.txt",
        "hist_feats_file": "./data/WikiQA/relation_test.binsum-20.txt"
    }
  },
  "outputs": {
    "predict": {
      "save_format": "TREC",
      "save_path": "predict.test.anmm.wikiqa.txt"
    }
  },
  "model": {
    "model_path": "./matchzoo/models/",
    "model_py": "anmm.ANMM",
    "setting": {
        "num_layers": 2,
        "hidden_sizes": [20, 1],
	    "dropout_rate": 0.0
    }
  },
  "losses": [
    {
       "object_name": "rank_hinge_loss" ,
       "object_params": {
            "margin": 1.0
       }
    }
  ],
  "metrics": [ "ndcg@3", "ndcg@5", "map" ]
}