visobert-multitask-2 / config.json
bie-nhd's picture
Upload folder using huggingface_hub
b3fdc92 verified
{
"model_name": "uitnlp/visobert",
"task_config": {
"sentiment": {
"dataset": "hung20gg/NEU-ESC",
"num_labels": 4,
"type": "single_label",
"text_col": "Text",
"label_col": "Sentiment"
},
"topic": {
"dataset": "hung20gg/NEU-ESC",
"num_labels": 10,
"type": "single_label",
"text_col": "Text",
"label_col": "Classification"
},
"hate_speech": {
"dataset": "sonlam1102/vithsd",
"num_labels": 5,
"type": "multi_label",
"text_col": "content",
"label_col": "labels",
"label_list": [
"individual",
"groups",
"religion/creed",
"race/ethnicity",
"politics"
]
},
"clickbait": {
"dataset": "blanatole/ViClickbait-2025",
"num_labels": 2,
"type": "single_label",
"text_col": "text",
"label_col": "label",
"dual_input": true
}
},
"max_seq_length": 128,
"hidden_dropout": 0.2,
"training_config": {
"num_epochs": 10,
"batch_size": 32,
"learning_rate": 2e-05,
"loss_weighting": "uncertainty",
"task_sampling": "proportional"
},
"final_results": {
"hate_speech_f1": 1.0,
"hate_speech_acc": 1.0,
"hate_speech_loss": 8.714258829154326e-09,
"clickbait_f1": 0.7949312440038375,
"clickbait_acc": 0.8245614035087719,
"clickbait_loss": 0.43148229067975824,
"sentiment_f1": 1.0,
"sentiment_acc": 1.0,
"sentiment_loss": 0.0,
"topic_f1": 1.0,
"topic_acc": 1.0,
"topic_loss": 5.103137090982615e-11
},
"best_scores": {
"hate_speech": 1.0,
"clickbait": 0.8027838536488368,
"sentiment": 1.0,
"topic": 1.0
}
}