|
{ |
|
"_name_or_path": "distilbert/distilroberta-base", |
|
"architectures": [ |
|
"RobertaForSequenceClassification" |
|
], |
|
"attention_probs_dropout_prob": 0.1, |
|
"bos_token_id": 0, |
|
"classifier_dropout": null, |
|
"eos_token_id": 2, |
|
"hidden_act": "gelu", |
|
"hidden_dropout_prob": 0.1, |
|
"hidden_size": 768, |
|
"id2label": { |
|
"0": "Loans", |
|
"1": "Digital Entertainment", |
|
"2": "Convenience Stores", |
|
"3": "Gas Stations", |
|
"4": "Third Party", |
|
"5": "Uncategorized", |
|
"6": "Internal Account Transfer", |
|
"7": "Shops", |
|
"8": "Transfer Debit", |
|
"9": "Restaurants", |
|
"10": "Utilities", |
|
"11": "ATM", |
|
"12": "Insurance", |
|
"13": "Supermarkets and Groceries", |
|
"14": "Department Stores", |
|
"15": "Transfer Credit", |
|
"16": "Bank Fees", |
|
"17": "Payroll", |
|
"18": "Arts and Entertainment", |
|
"19": "Transfer", |
|
"20": "Transfer Deposit", |
|
"21": "Gyms and Fitness Centers", |
|
"22": "Travel", |
|
"23": "Clothing and Accessories", |
|
"24": "Check Deposit", |
|
"25": "Service", |
|
"26": "Food and Beverage Services", |
|
"27": "Healthcare", |
|
"28": "Interest", |
|
"29": "Telecommunication Services", |
|
"30": "Tax Refund", |
|
"31": "Payment" |
|
}, |
|
"initializer_range": 0.02, |
|
"intermediate_size": 3072, |
|
"label2id": { |
|
"ATM": 11, |
|
"Arts and Entertainment": 18, |
|
"Bank Fees": 16, |
|
"Check Deposit": 24, |
|
"Clothing and Accessories": 23, |
|
"Convenience Stores": 2, |
|
"Department Stores": 14, |
|
"Digital Entertainment": 1, |
|
"Food and Beverage Services": 26, |
|
"Gas Stations": 3, |
|
"Gyms and Fitness Centers": 21, |
|
"Healthcare": 27, |
|
"Insurance": 12, |
|
"Interest": 28, |
|
"Internal Account Transfer": 6, |
|
"Loans": 0, |
|
"Payment": 31, |
|
"Payroll": 17, |
|
"Restaurants": 9, |
|
"Service": 25, |
|
"Shops": 7, |
|
"Supermarkets and Groceries": 13, |
|
"Tax Refund": 30, |
|
"Telecommunication Services": 29, |
|
"Third Party": 4, |
|
"Transfer": 19, |
|
"Transfer Credit": 15, |
|
"Transfer Debit": 8, |
|
"Transfer Deposit": 20, |
|
"Travel": 22, |
|
"Uncategorized": 5, |
|
"Utilities": 10 |
|
}, |
|
"layer_norm_eps": 1e-05, |
|
"max_position_embeddings": 514, |
|
"model_type": "roberta", |
|
"num_attention_heads": 12, |
|
"num_hidden_layers": 6, |
|
"pad_token_id": 1, |
|
"position_embedding_type": "absolute", |
|
"problem_type": "single_label_classification", |
|
"torch_dtype": "float32", |
|
"transformers_version": "4.44.2", |
|
"type_vocab_size": 1, |
|
"use_cache": true, |
|
"vocab_size": 50265 |
|
} |
|
|