minuva
/

MiniLMv2-goemotions-v2-onnx

Text Classification

multi-class-classification

multi-label-classification

Model card Files Files and versions Community

Ngit commited on Dec 22, 2023

Commit

e4454c6

·

1 Parent(s): 18a0550

Update README.md

Files changed (1) hide show

README.md +6 -7

README.md CHANGED Viewed

@@ -19,12 +19,12 @@ model-index:
     metrics:
     - name: F1
       type: f1
-      value: 0.487
 ---
 # Text Classification GoEmotions
-This a onnx quantized model ahd is fined-tuned version of [nreimers/MiniLMv2-L6-H384-distilled-from-RoBERTa-Large](https://huggingface.co/nreimers/MiniLMv2-L6-H384-distilled-from-RoBERTa-Large) on the on the [go_emotions](https://huggingface.co/datasets/go_emotions) dataset using [tasinho/text-classification-goemotions](https://huggingface.co/tasinhoque/text-classification-goemotions) as teacher model.
 # Load the Model
@@ -114,10 +114,9 @@ The following hyperparameters were used during training:
 | Teacher (params)    |   Student (params)     | Set         | Score (teacher)    |    Score (student)      |
 |--------------------|-------------|----------|--------| --------|
-| tasinhoque/text-classification-goemotions (355M) |      MiniLMv2-L6-H384-goemotions-v2    | Validation  | 0.514252 |0.484898 |
-| tasinhoque/text-classification-goemotions (33M) |      MiniLMv2-L6-H384-goemotions-v2 (original model)   | Test  | 0.501937 |  0.486890 |
-# Training Code, Evaluation & Deployment
-Check

     metrics:
     - name: F1
       type: f1
+      value: 0.482
 ---
 # Text Classification GoEmotions
+This a onnx quantized model and is fined-tuned version of [nreimers/MiniLMv2-L6-H384-distilled-from-RoBERTa-Large](https://huggingface.co/nreimers/MiniLMv2-L6-H384-distilled-from-RoBERTa-Large) on the on the [go_emotions](https://huggingface.co/datasets/go_emotions) dataset using [tasinho/text-classification-goemotions](https://huggingface.co/tasinhoque/text-classification-goemotions) as teacher model.
 # Load the Model
 | Teacher (params)    |   Student (params)     | Set         | Score (teacher)    |    Score (student)      |
 |--------------------|-------------|----------|--------| --------|
+| tasinhoque/text-classification-goemotions (355M) |      MiniLMv2-L6-H384-goemotions-v2-onnx    | Validation  | 0.514252 | .0478 |
+| tasinhoque/text-classification-goemotions (33M) |      MiniLMv2-L6-H384-goemotions-v2-onnx (original model)   | Test  | 0.501937 |  0.482 |
+# Deployment
+Check [this repository](https://github.com/minuva/emotion-prediction-serverless) to see how to easily deploy this model in a serverless environment with fast CPU inference.