detection/config.json

{
    "model_type": "bert-emotion-detection",
    "base_model": "bert-base-chinese",
    "architecture": "EmotionDetectionModel",
    "task": "emotion-classification",
    "num_emotions": 6,
    "emotion_labels": {
        "0": "sadness",
        "1": "joy",
        "2": "love",
        "3": "anger",
        "4": "fear",
        "5": "surprise"
    },
    "model_params": {
        "hidden_size": 768,
        "dropout": 0.1,
        "classifier_architecture": "two-layer-mlp",
        "intermediate_size": 384
    },
    "training_info": {
        "dataset": "dair-ai/emotion",
        "max_length": 512,
        "validation_accuracy": 0.9147,
        "framework": "pytorch"
    },
    "inference": {
        "max_length": 512,
        "batch_size": 16,
        "device": "cuda or cpu"
    },
    "version": "1.0",
    "created_by": "LongEmotion Detection Task"
}