RNN_test_Model / config.json
OneclickAI's picture
Upload 4 files
9a24bb1 verified
raw
history blame contribute delete
500 Bytes
{
"model_type": "SimpleRNN_for_Sequence_Classification",
"embedding_layer": {
"input_dim": 10000,
"output_dim": 32
},
"rnn_layer": {
"type": "SimpleRNN",
"units": 32
},
"classifier_head": {
"units": 1,
"activation": "sigmoid"
},
"preprocessing": {
"max_sequence_length": 256,
"padding_type": "pre"
},
"training_details": {
"loss_function": "binary_crossentropy",
"optimizer": "adam",
"metrics": ["accuracy"]
}
}