Question_Classifier_Pytorch/train_rnn_parameter.json at refactor · NTU-SCSE/Question_Classifier_Pytorch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
{
    "embedding_dim": 300,
    "hidden_dim": 256,
    "num_layers": 1,
    "dropout": 0.5,
    "zoneout": 0.5,
    "window": 2,
    "save_prev_x": false,
    "train_embedding_layer": false,
    "use_gpu": true,
    "epoch": 300,
    "batch_size": 64,
    "lr": 1e-3,
    "weight_decay": 5e-4,
    "max_text_length": null,
    "n_folds": 10,
    "train_dataset_path": "../dataset/faqs/faq_ntu_prototype_v6_train.csv",
    "dev_dataset_path": "../dataset/faqs/faq_ntu_prototype_v6_dev.csv",
    "use_git": true,
    "result_folder_path": "runs",
    "saved_model_file_path": "best_model/faq_best_model.model",
    "saved_text_pipeline_file_path": "preprocessing_pipeline/text_pipeline.pkl",
    "saved_label_pipeline_file_path": "preprocessing_pipeline/label_pipeline.pkl",
    "train_log_folder_path": "summaries/train",
    "dev_log_folder_path": "summaries/dev",
    "confusion_matrix_folder_path": "summaries/confusion_matrix",
    "pretrained_word_embedding_name": "glove.6B.300d",
    "pretrained_word_embedding_path": "../dataset/GoogleNews-vectors-negative300.bin"
}