AT707's picture
Initial release of InferenceGuard RoBERTa risk classifier
b445e0b verified
Raw History Blame Contribute Delete
442 Bytes
{
"architectures": [
"RobertaRiskClassifier"
],
"attributes": [
"age",
"location",
"occupation",
"education"
],
"base_model_name": "roberta-base",
"classifier_dropout": 0.1,
"dtype": "float32",
"hidden_size": 768,
"model_type": "inferenceguard-roberta",
"num_classes_per_head": 1,
"num_heads": 4,
"temperature_values": [
1.0,
1.0,
1.0,
1.0
],
"transformers_version": "5.20.2"
}