modernbert-nli-heads / config.json
jmccardle's picture
Upload folder using huggingface_hub
fa35377 verified
{
"base_model": "answerdotai/ModernBERT-large",
"nli_hidden_dim": 512,
"nli_classes": 3,
"abstention_hidden_dim": 128,
"abstention_classes": 2,
"nli_labels": [
"entailment",
"neutral",
"contradiction"
],
"abstention_labels": [
"confident",
"uncertain"
],
"training": {
"nli_epochs": 5,
"nli_accuracy": 0.708,
"abstention_epochs": 3,
"abstention_accuracy": 0.6546,
"abstention_recall": 0.766
}
}