File size: 803 Bytes
81412fa |
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 |
{
"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
} |