File size: 2,625 Bytes
2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d 3a2a4df 2759f8d |
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 |
{
"_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
} |