| { | |
| "dataset_reader": { | |
| "type": "classification-tsv", | |
| "token_indexers": { | |
| "tokens": { | |
| "type": "single_id" | |
| } | |
| } | |
| }, | |
| "model": { | |
| "type": "simple_classifier", | |
| "embedder": { | |
| "token_embedders": { | |
| "tokens": { | |
| "type": "embedding", | |
| "embedding_dim": 10 | |
| } | |
| } | |
| }, | |
| "encoder": { | |
| "type": "bag_of_embeddings", | |
| "embedding_dim": 10 | |
| } | |
| }, | |
| "train_data_path": "data/movie_review/train.tsv", | |
| "validation_data_path": "data/movie_review/dev.tsv", | |
| "trainer": { | |
| "callbacks": [ | |
| { | |
| "type": "tensorboard" | |
| } | |
| ], | |
| "num_epochs": 5, | |
| "optimizer": "adam" | |
| }, | |
| "data_loader": { | |
| "batch_size": 8, | |
| "shuffle": true | |
| } | |
| } |