| { | |
| "embedding_size": 768, | |
| "n_classes": 3, | |
| "model_name": "cnn_model", | |
| "optimizer": "Adam", | |
| "loss": "categorical_crossentropy", | |
| "learning_rate": 0.01, | |
| "learning_rate_decay": 0.1, | |
| "last_layer_activation": "softmax", | |
| "restore_lr": false, | |
| "classes": null, | |
| "text_size": null, | |
| "padding": "pre", | |
| "in": [ | |
| "word_emb" | |
| ], | |
| "in_y": [ | |
| "y_onehot" | |
| ], | |
| "out": [ | |
| "y_pred_probas" | |
| ], | |
| "main": true, | |
| "save_path": "/content/drive/MyDrive/BERT/custom_bert/model", | |
| "load_path": "/content/drive/MyDrive/BERT/custom_bert/model", | |
| "kernel_sizes_cnn": [ | |
| 3, | |
| 5, | |
| 7 | |
| ], | |
| "filters_cnn": 512, | |
| "coef_reg_cnn": 0.001, | |
| "coef_reg_den": 0.01, | |
| "dropout_rate": 0.5, | |
| "dense_size": 100, | |
| "mode": "infer", | |
| "final_learning_rate": 0.009999999776482582, | |
| "epochs_done": 0 | |
| } |