dream

1
{
2
    "dataset_reader": {
3
        "class_name": "basic_classification_reader",
4
        "x": [
5
            "context",
6
            "curr_utterance"
7
        ],
8
        "y": "final_label",
9
        "data_path": "{DOWNLOADS_PATH}/out_of_dialog_analysis/",
10
        "train": "train_ood_history_2_classes.csv",
11
        "valid": "valid_ood_history_2_classes.csv"
12
    },
13
    "dataset_iterator": {
14
        "class_name": "siamese_iterator",
15
        "seed": 243
16
    },
17
    "chainer": {
18
        "in": [
19
            "context",
20
            "curr_utterance"
21
        ],
22
        "in_y": [
23
            "y"
24
        ],
25
        "pipe": [
26
            {
27
                "class_name": "bert_preprocessor",
28
                "vocab_file": "{DOWNLOADS_PATH}/bert_models/uncased_L-12_H-768_A-12/vocab.txt",
29
                "do_lower_case": true,
30
                "max_seq_length": 64,
31
                "in": [
32
                    "context",
33
                    "curr_utterance"
34
                ],
35
                "out": [
36
                    "bert_features"
37
                ]
38
            },
39
            {
40
                "id": "classes_vocab",
41
                "class_name": "simple_vocab",
42
                "fit_on": [
43
                    "y"
44
                ],
45
                "save_path": "{MODEL_PATH}/classes.dict",
46
                "load_path": "{MODEL_PATH}/classes.dict",
47
                "in": [
48
                    "y"
49
                ],
50
                "out": [
51
                    "y_ids"
52
                ]
53
            },
54
            {
55
                "in": [
56
                    "y_ids"
57
                ],
58
                "out": [
59
                    "y_onehot"
60
                ],
61
                "class_name": "one_hotter",
62
                "depth": "#classes_vocab.len",
63
                "single_vector": true
64
            },
65
            {
66
                "class_name": "tf_breakdown_classifier",
67
                "n_classes": "#classes_vocab.len",
68
                "return_probas": true,
69
                "one_hot_labels": false,
70
                "bert_config_file": "{DOWNLOADS_PATH}/bert_models/uncased_L-12_H-768_A-12/bert_config.json",
71
                "pretrained_bert": "{DOWNLOADS_PATH}/bert_models/uncased_L-12_H-768_A-12/bert_model.ckpt",
72
                "save_path": "{MODEL_PATH}/model",
73
                "load_path": "{MODEL_PATH}/model",
74
                "keep_prob": 0.5,
75
                "learning_rate": 2e-05,
76
                "learning_rate_drop_patience": 2,
77
                "learning_rate_drop_div": 2.0,
78
                "in": [
79
                    "bert_features"
80
                ],
81
                "in_y": [
82
                    "y_ids"
83
                ],
84
                "out": [
85
                    "y_pred_probas"
86
                ]
87
            }
88
        ],
89
        "out": [
90
            "y_pred_probas"
91
        ]
92
    },
93
    "train": {
94
        "epochs": 100,
95
        "batch_size": 64,
96
        "metrics": [
97
            "f1_macro",
98
            "accuracy"
99
        ],
100
        "validation_patience": 5,
101
        "val_every_n_epochs": 1,
102
        "log_every_n_epochs": 1,
103
        "show_examples": false,
104
        "evaluation_targets": [
105
            "train",
106
            "valid"
107
        ],
108
        "class_name": "nn_trainer",
109
        "tensorboard_log_dir": "{MODEL_PATH}/"
110
    },
111
    "metadata": {
112
        "imports": [
113
            "breakdown_classifier"
114
        ],
115
        "variables": {
116
            "ROOT_PATH": "~/.deeppavlov",
117
            "DOWNLOADS_PATH": "{ROOT_PATH}/downloads",
118
            "MODELS_PATH": "{ROOT_PATH}/models",
119
            "MODEL_PATH": "{MODELS_PATH}/classifiers/ood_dbdc_bert_v1"
120
        },
121
        "requirements": [
122
            "{DEEPPAVLOV_PATH}/requirements/tf.txt",
123
            "{DEEPPAVLOV_PATH}/requirements/bert_dp.txt"
124
        ],
125
        "download": [
126
            {
127
                "url": "http://files.deeppavlov.ai/deeppavlov_data/bert/uncased_L-12_H-768_A-12.zip",
128
                "subdir": "{DOWNLOADS_PATH}/bert_models"
129
            },
130
            {
131
                "url": "http://files.deeppavlov.ai/deeppavlov_data/classifiers/ood_dbdc_bert_v1.tar.gz",
132
                "subdir": "{MODELS_PATH}/classifiers"
133
            }
134
        ]
135
    }
136
}
137