tool-call-verifier / training_config.json
Huamin's picture
Upload folder using huggingface_hub
83e80b5 verified
raw
history blame
554 Bytes
{
"model_name": "answerdotai/ModernBERT-base",
"num_labels": 3,
"label2id": {
"AUTHORIZED": 0,
"SUSPICIOUS": 1,
"UNAUTHORIZED": 2
},
"id2label": {
"0": "AUTHORIZED",
"1": "SUSPICIOUS",
"2": "UNAUTHORIZED"
},
"severity": {
"AUTHORIZED": 0,
"SUSPICIOUS": 2,
"UNAUTHORIZED": 4
},
"loss": "CrossEntropyLoss (class-weighted)",
"class_weights": [
0.5,
2.0,
3.0
],
"optimization_target": "unauthorized_f1",
"batch_size": 4,
"epochs": 6,
"learning_rate": 1e-05,
"max_length": 2048
}