Views
No views yet
pip install model2vec[inference]1from model2vec.inference import StaticModelPipeline
2
3model = StaticModelPipeline.from_pretrained(
4 "enguard/tiny-guard-2m-en-prompt-harassment-binary-moderation"
5)
6
7
8# Supports single texts. Format input as a single text:
9text = "Example sentence"
10
11model.predict([text])
12model.predict_proba([text])
13| Field | Value |
|---|---|
| Classifies | prompt-harassment-binary |
| Base Model | minishlab/potion-base-2m |
| Precision | 0.8788 |
| Recall | 0.7180 |
| F1 | 0.7903 |
| True \ Predicted | FAIL | PASS |
|---|---|---|
| FAIL | 1421 | 563 |
| PASS | 196 | 1788 |
1{
2 "FAIL": {
3 "precision": 0.8787878787878788,
4 "recall": 0.7180394138453765,
5 "f1-score": 0.7903225806451613,
6 "support": 1979.0
7 },
8 "PASS": {
9 "precision": 0.7605150214592274,
10 "recall": 0.9004065040650406,
11 "f1-score": 0.8245695672405771,
12 "support": 1968.0
13 },
14 "accuracy": 0.8089688370914618,
15 "macro avg": {
16 "precision": 0.8196514501235531,
17 "recall": 0.8092229589552086,
18 "f1-score": 0.8074460739428692,
19 "support": 3947.0
20 },
21 "weighted avg": {
22 "precision": 0.8198162590202613,
23 "recall": 0.8089688370914618,
24 "f1-score": 0.8073983520208335,
25 "support": 3947.0
26 }
27}| Text | True Label | Predicted Label |
|---|---|---|
| you don't have much experience with european gypsies, do you? Touch one, and they spawn cousins all around you like an RPG boss. | FAIL | FAIL |
| 良い、悪天候時butthole! | FAIL | FAIL |
| I know what you are thinking....but where will he build the fire for the smoke signals? | PASS | PASS |
| Not really even an "unnatural" disaster. Oil comes from the earth, water comes from the earth. Its inevitable that they will find each other. | PASS | PASS |
| You don't even know what you were quoting, do you? You just saw it on the internet somewhere and decided to parrot it here on reddit for karma. Nicely done, I'll give you a C for effort, but you should try harder next time. | FAIL | FAIL |
| you don't have much experience with european gypsies, do you? Touch one, and they spawn cousins all around you like an RPG boss. | FAIL | FAIL |
| Dataset Size | Time (seconds) | Predictions/Second |
|---|---|---|
| 1 | 0.0002 | 4928.68 |
| 1000 | 0.0504 | 19859.39 |
| 3968 | 0.2013 | 19709.62 |
@software{minishlab2024model2vec,
author = {Stephan Tulkens and {van Dongen}, Thomas},
title = {Model2Vec: Fast State-of-the-Art Static Embeddings},
year = {2024},
publisher = {Zenodo},
doi = {10.5281/zenodo.17270888},
url = {https://github.com/MinishLab/model2vec},
license = {MIT}
}