bot_kvret_infer.json

{
  "chainer": {
    "in": ["x_text"],
    "in_y": ["y_text", "y_domain"],
    "out": ["prediction_text"],
    "pipe": [
      {
        "id": "stream_spacy",
        "name": "stream_spacy_tokenizer",
        "alphas_only": false,
        "in": ["x_text"],
        "out": ["x_tokens"]
      },
      {
        "ref": "stream_spacy",
        "in": ["y_text"],
        "out": ["y_tokens"]
      },
      {
        "id": "kb",
        "name": "knowledge_base",
        "tokenizer": "#stream_spacy",
        "save_path": "seq2seq_go_bot/kvret_kb.json",
        "load_path": "seq2seq_go_bot/kvret_kb.json"
      },
      {
        "id": "src_token_vocab",
        "name": "default_vocab",
        "level": "token",
        "special_tokens": ["<SOS>"],
        "save_path": "vocabs/src_tokens.dict",
        "load_path": "vocabs/src_tokens.dict"
      },
      {
        "id": "tgt_token_vocab",
        "name": "default_vocab",
        "level": "token",
        "special_tokens": ["<SOS>", "<EOS>"],
        "save_path": "vocabs/tgt_tokens.dict",
        "load_path": "vocabs/tgt_tokens.dict"
      },
      {
        "in": ["x_tokens"],
        "in_y": ["y_tokens"],
        "out": ["prediction_tokens"],
        "main": true,
        "name": "seq2seq_go_bot",
        "start_of_sequence_token": "<SOS>",
        "end_of_sequence_token": "<EOS>",
        "network": {
          "name": "seq2seq_go_bot_nn",
          "load_path": "seq2seq_go_bot/model",
          "save_path": "seq2seq_go_bot/model",
          "learning_rate": 0.0009,
          "target_start_of_sequence_index": "#tgt_token_vocab.__getitem__('<SOS>')",
          "target_end_of_sequence_index": "#tgt_token_vocab.__getitem__('<EOS>')",
          "source_vocab_size": "#src_token_vocab.__len__()",
          "target_vocab_size": "#tgt_token_vocab.__len__()",
          "hidden_size": 256
        },
        "source_vocab": "#src_token_vocab",
        "target_vocab": "#tgt_token_vocab",
        "debug": false
      },
      {
        "ref": "stream_spacy",
        "in": ["prediction_tokens"],
        "out": ["prediction_text"]
      }
    ]
  },
  "train": {
    "epochs": 200,
    "batch_size": 16,

    "metrics": ["per_item_bleu", "per_item_accuracy"],
    "validation_patience": 20,
    "val_every_n_epochs": 1,

    "log_every_n_batches": -1,
    "log_every_n_epochs": 1,
    "show_examples": false
  },
  "metadata": {
    "labels": {
      "telegram_utils": "Seq2SeqGoalOrientedBot",
      "server_utils": "GoalOrientedBot"
    },
    "download": [
      "http://lnsigo.mipt.ru/export/deeppavlov_data/vocabs.tar.gz",
      "http://lnsigo.mipt.ru/export/deeppavlov_data/seq2seq_go_bot.tar.gz"
    ]
  }
}