File size: 755 Bytes
7ec1e86 f22eaf8 975257f f22eaf8 7ec1e86 f22eaf8 |
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 |
{
"model_type": "custom",
"architectures": ["LSTM"],
"library_name": "tensorflow",
"task_specific_params": {
"text-classification": {
"vocab_size": 500,
"embedding_dim": 64,
"hidden_size": 64,
"num_layers": 2,
"dropout_rate": 0.2,
"max_sequence_length": 10
}
},
"training_params": {
"batch_size": 8,
"epochs": 50,
"loss_function": "sparse_categorical_crossentropy",
"optimizer": "adam",
"metrics": ["accuracy"]
},
"train_data_size": 5000,
"id2label": {
"0": "Lifestyle",
"1": "Movies & Entertainment",
"2": "Food & Dining",
"3": "Others"
},
"label2id": {
"Lifestyle": 0,
"Movies & Entertainment": 1,
"Food & Dining": 2,
"Others": 3
}
} |