allennlp
28 строк · 790.0 Байт
1local COMMON = import 'common.jsonnet';
2
3{
4"dataset_reader": COMMON['dataset_reader'],
5"datasets_for_vocab_creation": ["train"],
6"train_data_path": COMMON['train_data_path'],
7"validation_data_path": COMMON['train_data_path'],
8"model": {
9"type": "basic_classifier",
10"text_field_embedder": {
11"token_embedders": {
12"tokens": {
13"type": "embedding",
14"embedding_dim": 10,
15"trainable": true
16}
17}
18},
19"seq2vec_encoder": {
20"type": "cnn",
21"num_filters": 8,
22"embedding_dim": 10,
23"output_dim": 16
24}
25},
26"data_loader": COMMON['data_loader'],
27"trainer": COMMON['trainer'],
28}
29