Добавлен Model Card с метриками и примером использования
Browse files
README.md
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
language:
|
|
@@ -43,9 +44,12 @@ tokenizer = AutoTokenizer.from_pretrained("NikolayRainWay/multitask-toxic-encode
|
|
| 43 |
model = MultiTaskToxicityEncoder()
|
| 44 |
model.load_state_dict(torch.load("pytorch_model.bin", map_location='cpu'))
|
| 45 |
model.eval()
|
| 46 |
-
|
| 47 |
Автор
|
|
|
|
| 48 |
NikolayRainWay (в рамках учебного курса по трансформерным энкодерам)
|
| 49 |
|
| 50 |
Датасет
|
| 51 |
Обучен на датасете NikolayRainWay/multitask-very-toxic-comments
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
---
|
| 3 |
license: mit
|
| 4 |
language:
|
|
|
|
| 44 |
model = MultiTaskToxicityEncoder()
|
| 45 |
model.load_state_dict(torch.load("pytorch_model.bin", map_location='cpu'))
|
| 46 |
model.eval()
|
|
|
|
| 47 |
Автор
|
| 48 |
+
---
|
| 49 |
NikolayRainWay (в рамках учебного курса по трансформерным энкодерам)
|
| 50 |
|
| 51 |
Датасет
|
| 52 |
Обучен на датасете NikolayRainWay/multitask-very-toxic-comments
|
| 53 |
+
|
| 54 |
+
Лицензия
|
| 55 |
+
MIT
|