allen_conf.json

{
  "dataset_reader": {
    "type": "augmented_mention_categories",
    "category_mapping_file": "./data/category_mapping.json",
    "tokenizer": {
      "type": "word",
      "word_splitter": {
        "type": "fast-splitter"
      }
    },
    "token_indexers": {
      "tokens": {
        "type": "fasttext-id",
        "model_path": "./data/fasttext_embedding.model",
        "model_params_path": "./data/fasttext_embedding.model.params"
      }
    }
  },
  "train_data_path": "./data/train_data.tsv",
  "validation_data_path": "./data/valid_data.tsv",
  "evaluate_on_test": true,
  "model": {
    "type": "category_predictor",
    "text_embedder": {
      "type": "basic",
      "allow_unmatched_keys": true,
      "embedder_to_indexer_map": {
        "tokens-ngram": [
          "tokens-ngram",
          "tokens-ngram-lengths",
          "tokens-ngram-mask"
        ],
        "tokens": [
          "tokens"
        ]
      },
      "token_embedders": {
        "tokens-ngram": {
          "type": "fasttext-embedder",
          "model_path": "./data/fasttext_embedding.model",
          "trainable": false
        },
        "tokens": {
          "type": "embedding",
          "embedding_dim": 20,
          "trainable": true
        }
      }
    },
    "encoder": {
      "type": "lstm",
      "input_size": 320,
      "hidden_size": 100,
      "num_layers": 1,
      "dropout": 0.2,
      "bidirectional": true
    },
    "seq_combiner": {
      "type": "attention-combiner",
      "num_layers": 3,
      "input_state_dim": 200,
      "encoded_seq_dim": 520,
      "feed_forward_hidden_dim": 200,
      "hidden_dim": 200,
      "output_dim": 100,
      "num_heads": 5
    }
  },
  "iterator": {
    "type": "numpy-multiprocessing",
    "num_workers": 7,
    "base_iterator": {
      "type": "basic",
      "batch_size": 32
    }
  },
  "trainer": {
    "shuffle": false,
    "optimizer": {
      "type": "adam",
      "lr": 0.001
    },
    "should_log_learning_rate": true,
    "num_epochs": 200,
    "learning_rate_scheduler": {
      "type": "reduce_on_plateau",
      "factor": 0.5,
      "patience": 2,
      "verbose": true
    }
  },
  "vocabulary": {
    "directory_path": "./data/vocabulary/"
  }
}