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