deepx-arabic-absa-marbert / model_config.json
Dohahemdann's picture
Upload model_config.json with huggingface_hub
6a27fba verified
{
"model_name": "UBC-NLP/MARBERTv2",
"architecture": "AspectSentimentModel",
"num_aspects": 9,
"num_sentiment_classes": 3,
"dropout_rate": 0.15,
"hidden_dim": 384,
"mean_pool": true,
"max_length": 256,
"neutral_margin": 0.08,
"neutral_boost": 5.0,
"aspect_names": [
"food",
"service",
"price",
"cleanliness",
"delivery",
"ambiance",
"app_experience",
"general",
"none"
],
"sentiment_map": {
"positive": 0,
"negative": 1,
"neutral": 2
},
"best_joint_f1": 0.7831,
"best_threshold": 0.55,
"best_epoch": 8,
"stage": "STAGE 4 — Neutral consolidation (encoder frozen)",
"val_metrics": {
"aspect_f1": 0.8611,
"aspect_precision": 0.8776,
"aspect_recall": 0.8452,
"sentiment_f1": 0.9113,
"sentiment_precision": 0.9113,
"sentiment_recall": 0.9113,
"joint_f1": 0.7847,
"joint_precision": 0.7998,
"joint_recall": 0.7702,
"aspect_per_class": {
"food": {
"f1": 0.8911,
"precision": 0.9,
"recall": 0.8824
},
"service": {
"f1": 0.8947,
"precision": 0.917,
"recall": 0.8735
},
"price": {
"f1": 0.8333,
"precision": 0.8553,
"recall": 0.8125
},
"cleanliness": {
"f1": 0.8247,
"precision": 0.8696,
"recall": 0.7843
},
"delivery": {
"f1": 0.8043,
"precision": 0.8409,
"recall": 0.7708
},
"ambiance": {
"f1": 0.8105,
"precision": 0.8556,
"recall": 0.77
},
"app_experience": {
"f1": 0.8954,
"precision": 0.8992,
"recall": 0.8917
},
"general": {
"f1": 0.8333,
"precision": 0.7927,
"recall": 0.8784
},
"none": {
"f1": 0.6957,
"precision": 0.7273,
"recall": 0.6667
}
},
"sentiment_per_class": {
"positive": {
"f1": 0.932,
"precision": 0.9332,
"recall": 0.9308
},
"negative": {
"f1": 0.9202,
"precision": 0.9033,
"recall": 0.9377
},
"neutral": {
"f1": 0.5,
"precision": 0.619,
"recall": 0.4194
}
},
"val_loss": 2.8323
}
}