Downloads · 30 days
12
0% of all-time downloads
pablobonfim13/Distilbert-atis-intent-classifier-Classificador-de-intencao-ML
Distilbert-atis-intent-classifier-Classificador-de-intencao-ML is a text classification model from pablobonfim13. Use it when you need a label for a piece of text. The card lists the license as mit.
Este é um modelo DistilBERT (distilbert-base-uncased) que foi adaptado (fine-tuned) para a tarefa de Classificação de Intenção usando o dataset ATIS.
Downloads · 30 days
12
0% of all-time downloads
All-time downloads
6K
Public
Parameters
67M
268 MB on disk
Likes
1
Public
Click a slice to open those files.
.safetensors536 MB · 100%
From the Hugging Face model README
Este é um modelo DistilBERT (distilbert-base-uncased) que foi adaptado (fine-tuned) para a tarefa de Classificação de Intenção usando o dataset ATIS.
Este projeto foi desenvolvido como parte de um estudo prático sobre o pipeline de NLP (Processamento de Linguagem Natural), desde a teoria (baseado no artigo "Attention Is All You Need") até a produção de um modelo funcional.
O modelo é capaz de classificar 26 intenções diferentes relacionadas a consultas de voos.
O modelo foi treinado por 3 épocas e atingiu os seguintes resultados no conjunto de testes (dados nunca vistos):
Este resultado de alta acurácia prova que o modelo é robusto e confiável para rotear solicitações de usuários.
Você pode usar este modelo diretamente no seu próprio código Python com a biblioteca transformers:
from transformers import pipeline
# Use o nome do SEU repositório aqui
model_path = "user/repo_path"
# Carrega o pipeline
pipe = pipeline("text-classification", model=model_path)
# Faz a previsão
resultado = pipe("i need a flight to new york")
print(resultado)
# Saída esperada: [{'label': 'flight', 'score': 0.999...}]
resultado = pipe("what is the abbreviation for denver")
print(resultado)
# Saída esperada: [{'label': 'abbreviation', 'score': 0.927...}]