Views
No views yet
| Metric | Train | Validation |
|---|---|---|
| Loss | 0.123 | 0.156 |
| Accuracy | 0.956 | 0.932 |
| F1 | 0.953 | 0.928 |
| Metric | Value |
|---|---|
| Accuracy | 0.925 |
| Precision | 0.928 |
| Recall | 0.923 |
| F1 | 0.925 |
1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2import torch
3
4model = AutoModelForSequenceClassification.from_pretrained("miguelmejias0512/deepseek-solidity-coder-llm-7b-finetuned")
5tokenizer = AutoTokenizer.from_pretrained("miguelmejias0512/deepseek-solidity-coder-llm-7b-finetuned")
6
7# Ejemplo de inferencia
8text = "Texto de ejemplo a clasificar"
9inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=512)
10
11with torch.no_grad():
12 outputs = model(**inputs)
13 predictions = torch.softmax(outputs.logits, dim=-1)
14
15print(f"Probabilidades por clase: {predictions.numpy()}")
16Preprocesamiento necesario
17python
18def preprocess_text(text):
19 # Implementa aquí el mismo preprocesamiento usado en entrenamiento
20 text = text.lower()
21 # ... más pasos
22 return text
23
24# Uso completo con preprocesamiento
25raw_text = "TEXTO Crudo con MAYÚSCULAS y signos!"
26processed_text = preprocess_text(raw_text)
27inputs = tokenizer(processed_text, return_tensors="pt", truncation=True, padding=True)
28# ... resto del código de inferencia
29Ejemplo completo con post-procesamiento
30python
31import numpy as np
32
33class_names = ["clase_A", "clase_B", "clase_C"] # Reemplaza con tus clases
34
35def predict(text):
36 # Preprocesamiento
37 processed_text = preprocess_text(text)
38
39 # Tokenización
40 inputs = tokenizer(processed_text, return_tensors="pt", truncation=True, padding=True, max_length=512)
41
42 # Inferencia
43 with torch.no_grad():
44 outputs = model(**inputs)
45 probs = torch.softmax(outputs.logits, dim=-1).numpy()[0]
46
47 # Resultado
48 return {class_names[i]: float(probs[i]) for i in range(len(class_names))}
49
50# Ejemplo de uso
51result = predict("Este es un texto de ejemplo para clasificar")
52print(result)