config.json

{
  "LOAD_SAVE_IGNORES": [],
  "_commit_hash": null,
  "aggregator": null,
  "architectures": [
    "RecBot"
  ],
  "datahandler": null,
  "generator": {
    "LOAD_SAVE_IGNORES": [],
    "_commit_hash": null,
    "bos_token_id": 1,
    "context_size": 256,
    "eos_token_id": 2,
    "hidden_size": 256,
    "moduleName": "SwitchingDecoder",
    "moduleType": "generator",
    "num_layers": 1,
    "output_map": null,
    "peephole": false,
    "task_embedding.dtype": "torch.float32",
    "task_embedding.shape": [
      15005,
      300
    ],
    "transformers_version": null
  },
  "manager": {
    "LOAD_SAVE_IGNORES": [],
    "_commit_hash": null,
    "conversation_encoder_hidden_size": 256,
    "moduleName": "RedialManager",
    "moduleType": "manager",
    "n_movies": 6924,
    "num_vocab": 15005,
    "output_map": null,
    "transformers_version": null
  },
  "processor": [
    {
      "LOAD_SAVE_IGNORES": [],
      "_commit_hash": null,
      "conv_bidrectional": false,
      "conversation_encoder_hidden_size": 256,
      "conversation_encoder_num_layers": 1,
      "moduleName": "HRNN",
      "moduleType": "processor",
      "output_map": null,
      "sentence_encoder_hidden_size": 256,
      "sentence_encoder_model": "princeton-nlp/unsup-simcse-roberta-base",
      "sentence_encoder_num_layers": 1,
      "transformers_version": null,
      "use_dropout": false,
      "use_movie_occurrences": false
    },
    {
      "LOAD_SAVE_IGNORES": [],
      "_commit_hash": null,
      "_name_or_path": "ckpt/sa/model_best",
      "hrnn_params": {
        "conv_bidrectional": false,
        "conversation_encoder_hidden_size": 512,
        "conversation_encoder_num_layers": 2,
        "sentence_encoder_hidden_size": 512,
        "sentence_encoder_model": "princeton-nlp/unsup-simcse-roberta-base",
        "sentence_encoder_num_layers": 2,
        "use_dropout": 0.4,
        "use_movie_occurrences": "word"
      },
      "moduleName": "HRNNForClassification",
      "moduleType": "processor",
      "multiple_items_per_example": true,
      "output_classes": {
        "i_liked": 3,
        "i_seen": 3,
        "i_suggested": 1,
        "r_liked": 3,
        "r_seen": 3,
        "r_suggested": 1
      },
      "output_map": null,
      "return_liked_probability": true,
      "transformers_version": null
    }
  ],
  "recommender": {
    "LOAD_SAVE_IGNORES": [],
    "_commit_hash": null,
    "_name_or_path": "ckpt/autorec/db_model_best",
    "architectures": [
      "AutoRec"
    ],
    "f": "sigmoid",
    "g": "sigmoid",
    "layer_sizes": [
      1000
    ],
    "moduleName": "AutoRec",
    "moduleType": "recommender",
    "n_movies": 6924,
    "output_map": null,
    "torch_dtype": "float32",
    "transformers_version": null
  },
  "torch_dtype": "float32",
  "transformers_version": null
}