--- license: apache-2.0 metrics: - accuracy --- ## Modelo Base: [nicholasKluge/TeenyTinyLlama-160m](https://huggingface.co/nicholasKluge/TeenyTinyLlama-160m) ## Detalhes - **Epochs:** 3 - **Batch:** 16 - **Optimizer:** `torch.optim.AdamW` (learning_rate = 4e-5, epsilon = 1e-8) - **GPU:** 1 NVIDIA RTX 3060 - **TEMPO:** 36 minutos Execução: ```python from transformers import AutoTokenizer from transformers import AutoModelForSequenceClassification import torch tokenizer = AutoTokenizer.from_pretrained("JJhooww/Classificador_de_prompt") features = tokenizer(['Preciso de um script em python'], padding=True, truncation=True, return_tensors="pt") model = AutoModelForSequenceClassification.from_pretrained("JJhooww/Classificador_de_prompt") with torch.no_grad(): logits = model(**features).logits predicted_class_id = logits.argmax().item() model.config.id2label[predicted_class_id] #'Programação e Desenvolvimento' ``` ## Categorias de prompts ```markdown ### Informação Geral ### Análise e Interpretação ### Aprendizado e Tutorial ### Cálculos e Matemática ### Programação e Desenvolvimento ### Linguagem e Tradução ### Consultoria e Estratégia ### Criação e Criatividade ### Suporte e Ajuda ### Verificação e Validação ``` Accuracy: 0.791537%