Downloads · 30 days
5
5% of all-time downloads
kanowest/beto-steam-reviews
beto-steam-reviews is a text classification model from kanowest. 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.
Este modelo es una versión ajustada (fine-tuned) del modelo BETO (BERT en español).
Downloads · 30 days
5
5% of all-time downloads
All-time downloads
100
Public
Parameters
110M
439 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors439 MB · 100%
From the Hugging Face model README
Este modelo es una versión ajustada (fine-tuned) del modelo BETO (BERT en español).
Ha sido entrenado para clasificar reseñas de videojuegos de la plataforma Steam en dos categorías:
El objetivo era adaptar un modelo de lenguaje generalista (BETO) al lenguaje específico y coloquial de los "gamers" en español. El modelo ha sido entrenado con un dataset de aproximadamente 1.000 reseñas reales del juego Counter Strike 2. Por ello, es posible que a la hora de ser probado en juegos de distinto género, no funcione del todo bien.
El modelo alcanzó los siguientes resultados durante el entrenamiento:
Puedes usar este modelo directamente con la librería transformers, además de tener instalado torch. También se recomienda, aunque no es necesario, instalar accelerate y numpy:
from transformers import pipeline
# Cargar el pipeline
clasificador = pipeline("text-classification", model="kanowest/beto-steam-reviews")
# Probar con una frase
texto = "El juego tiene buenos gráficos pero la historia es aburrida."
resultado = clasificador(texto)
print(resultado)
# Salida esperada: [{'label': 'LABEL_0', 'score': 0.99...}]
Hice una prueba con un dataset del Team Fortress 2 y no salió todo lo bien que me esperaba. Falla bastante a la hora de reconocer reseñas negativas (más de lo que me imaginaba que fallaría). Es mi primer modelo y, por tanto, no ha salido demasiado bien. Mi plan es crear uno que funcione mejor en el futuro.