config.json

{
  "model_type": "sentiment-head",
  "architectures": [
    "TextEmbedding3SmallSentimentHead"
  ],
  "input_dim": 1536,
  "hidden_dim": 256,
  "dropout": 0.2,
  "num_labels": 3,
  "id2label": {
    "0": "negative",
    "1": "neutral",
    "2": "positive"
  },
  "label2id": {
    "negative": 0,
    "neutral": 1,
    "positive": 2
  },
  "auto_map": {
    "AutoConfig": "modeling_te3s_head.TextEmbedding3SmallSentimentHeadConfig",
    "AutoModel": "modeling_te3s_head.TextEmbedding3SmallSentimentHead"
  }
}