File size: 2,826 Bytes
f7a6247 2d5a8a0 f7a6247 8def456 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 2d5a8a0 f7a6247 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 |
{
"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
} |