| { |
| "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 |
| } |
