Modelo Dummy de DistilBert para Análisis de Sentimientos
dhpollack
Clasificación de texto
Este es un modelo dummy que puede ser utilizado para probar el pipeline de transformers con la tarea de análisis de sentimientos. Siempre debería dar resultados aleatorios (i.e. {"label": "negative", "score": 0.5}).
Como usar
classifier = pipeline("sentiment-analysis", "dhpollack/distilbert-dummy-sentiment")
results = classifier(["this is a test", "another test"])
import transformers
config = transformers.DistilBertConfig(vocab_size=5, n_layers=1, n_heads=1, dim=1, hidden_dim=4 * 1, num_labels=2, id2label={0: "negative", 1: "positive"}, label2id={"negative": 0, "positive": 1})
model = transformers.DistilBertForSequenceClassification(config)
tokenizer = transformers.DistilBertTokenizer("/tmp/vocab.txt", model_max_length=512)
config.save_pretrained(".")
model.save_pretrained(".")
tokenizer.save_pretrained(".")
Funcionalidades
- Clasificación de Texto
- Transformers
- PyTorch
- Multilingüe
- Inglés
- DistilBert
- Análisis de Sentimientos
- Pruebas de Unidad
- Puntos de Inferencia
Casos de uso
- Prueba del pipeline de transformers para la tarea de análisis de sentimientos.