mbti_bilstm_personality_prediction / ensemble_config.json
MalekOthman's picture
Upload ensemble_config.json with huggingface_hub
d7d3141 verified
Raw History Blame Contribute Delete
407 Bytes
{
"ensemble_method": "average",
"model_architecture": {
"type": "bilstm",
"layers": 2,
"hidden_dim": 192
},
"component_models": [
"fold_0/pytorch_model.bin",
"fold_1/pytorch_model.bin",
"fold_2/pytorch_model.bin",
"fold_3/pytorch_model.bin",
"fold_4/pytorch_model.bin"
],
"preprocessing": {
"text_cleaning": true,
"max_length": 512
}
}