config.json

{
  "_commit_hash": null,
  "aggregator": null,
  "architectures": [
    "Pipeline"
  ],
  "datahandler": null,
  "generator": {
    "_commit_hash": null,
    "architectures": [
      "SwitchingDecoder"
    ],
    "context_size": 256,
    "embedding_path": "embeddings/glove.840B.300d.h5",
    "hidden_size": 256,
    "moduleName": "SwitchingDecoder",
    "moduleType": "generator",
    "num_layers": 1,
    "peephole": false,
    "torch_dtype": "float32",
    "transformers_version": null
  },
  "manager": {
    "_commit_hash": null,
    "conversation_encoder_hidden_size": 256,
    "moduleName": "RedialManager",
    "moduleType": "manager",
    "recommend_new_movies": true,
    "transformers_version": null
  },
  "processor": [
    {
      "_commit_hash": null,
      "architectures": [
        "HRNN"
      ],
      "conv_bidirectional": false,
      "conv_bidrectional": false,
      "conversation_encoder_hidden_size": 256,
      "conversation_encoder_num_layers": 1,
      "moduleName": "HRNN",
      "moduleType": "processor",
      "sentence_encoder_hidden_size": 256,
      "sentence_encoder_model": "princeton-nlp/unsup-simcse-roberta-base",
      "sentence_encoder_num_layers": 1,
      "torch_dtype": "float32",
      "transformers_version": null,
      "use_dropout": false,
      "use_movie_occurrences": false
    },
    {
      "_commit_hash": null,
      "architectures": [
        "HRNNForClassification"
      ],
      "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": false,
      "output_classes": {
        "i_liked": 3,
        "i_seen": 3,
        "i_suggested": 1,
        "r_liked": 3,
        "r_seen": 3,
        "r_suggested": 1
      },
      "return_liked_probability": false,
      "torch_dtype": "float32",
      "transformers_version": null
    }
  ],
  "recommender": {
    "_commit_hash": null,
    "architectures": [
      "AutoRec"
    ],
    "f": "sigmoid",
    "g": "sigmoid",
    "layer_sizes": [
      1000
    ],
    "moduleName": "AutoRec",
    "moduleType": "recommender",
    "n_movies": 6924,
    "torch_dtype": "float32",
    "transformers_version": null
  },
  "torch_dtype": "float32",
  "transformers_version": null
}