Downloads · 30 days
18
3% of all-time downloads
diwank/maptask-deberta-pair
maptask-deberta-pair is a text classification model from diwank. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
Deberta-based Daily MapTask style dialog-act annotations classification model
Downloads · 30 days
18
3% of all-time downloads
All-time downloads
622
Public
Repo size
2.8 GB
Likes
2
Public
Click a slice to open those files.
.pt1.1 GB · 50%
From the Hugging Face model README
Deberta-based Daily MapTask style dialog-act annotations classification model
from simpletransformers.classification import (
ClassificationModel, ClassificationArgs
)
model = ClassificationModel("deberta", "diwank/maptask-deberta-pair")
predictions, raw_outputs = model.predict([["Say what is the meaning of life?", "I dont know"]])
convert_to_label = lambda n: ["acknowledge (0), align (1), check (2), clarify (3), explain (4), instruct (5), query_w (6), query_yn (7), ready (8), reply_n (9), reply_w (10), reply_y (11)".split(', ')[i] for i in n]
convert_to_label(predictions) # reply_n (9)