Update README.md
Browse files
README.md
CHANGED
@@ -6,6 +6,7 @@ language:
|
|
6 |
# Text Classification GoEmotions
|
7 |
|
8 |
This model is a fined-tuned version of [nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large](https://huggingface.co/nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large) on the on the [Jigsaw 1st Kaggle competition](https://www.kaggle.com/competitions/jigsaw-toxic-comment-classification-challenge) dataset using [unitary/toxic-bert](https://huggingface.co/unitary/toxic-bert) as teacher model.
|
|
|
9 |
|
10 |
# Load the Model
|
11 |
|
@@ -37,5 +38,5 @@ The following hyperparameters were used during training:
|
|
37 |
|
38 |
# Deployment
|
39 |
|
40 |
-
Check this [repository](https://github.com/minuva/toxicity-prediction-serverless) to see how to deploy in a
|
41 |
|
|
|
6 |
# Text Classification GoEmotions
|
7 |
|
8 |
This model is a fined-tuned version of [nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large](https://huggingface.co/nreimers/MiniLMv2-L6-H384-distilled-from-BERT-Large) on the on the [Jigsaw 1st Kaggle competition](https://www.kaggle.com/competitions/jigsaw-toxic-comment-classification-challenge) dataset using [unitary/toxic-bert](https://huggingface.co/unitary/toxic-bert) as teacher model.
|
9 |
+
The quantized version in ONNX format can be found [here](minuva/MiniLMv2-toxic-jijgsaw-onnx). The model with two labels only (toxicity and severe toxicity) is [here](minuva/MiniLMv2-toxic-jijgsaw-lite-onnx)
|
10 |
|
11 |
# Load the Model
|
12 |
|
|
|
38 |
|
39 |
# Deployment
|
40 |
|
41 |
+
Check this [repository](https://github.com/minuva/toxicity-prediction-serverless) to see how to easily deploy this model in a serverless environment with fast CPU inference and light resource utilization.
|
42 |
|