banking-lm-working / config.json
rawsun00001's picture
Upload working banking LM - Training successful!
4413af1 verified
{
"model_type": "banking_lm_working",
"vocab_size": 61,
"hidden_size": 64,
"num_layers": 2,
"categories": [
"Automobile and Transportation",
"Food and Drinks",
"Shopping",
"Entertainment",
"Health and Wellness",
"Bills",
"Mobile",
"Cash Transactions",
"Salary",
"Reversal and Refunds",
"Self Transfer"
],
"training_loss": 0.94,
"task": "banking_sms_parsing",
"version": "1.0"
}