|
--- |
|
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% |
