| { | |
| "WEIGHT_DIMENSIONS": {}, | |
| "architectures": [ | |
| "RedialRec" | |
| ], | |
| "autorec_params": { | |
| "f": "sigmoid", | |
| "g": "sigmoid", | |
| "layer_sizes": [ | |
| 1000 | |
| ] | |
| }, | |
| "sa_params": { | |
| "hrnn_params": { | |
| "conv_bidirectional": 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" | |
| }, | |
| "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 | |
| }, | |
| "return_liked_probability": true | |
| }, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.33.2" | |
| } | |