Uso del modello fine-tunato
Il modello è stato fine-tunato. Ora sei pronto a usarlo su nuovi dati e generare delle classificazioni. Vediamo quanto bene il tuo modello fine-tunato riesce a etichettare nuove interazioni come a basso o alto rischio di churn.
Il tuo model fine-tunato e il tokenizer sono già stati caricati per te.
Questo esercizio fa parte del corso
Introduzione agli LLM in Python
Istruzioni dell'esercizio
- Tokenizza i nuovi dati.
- Passa gli input tokenizzati al modello fine-tunato, disattivando i gradienti.
- Estrai le nuove predizioni.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
input_text = ["I'd just like to say, I love the product! Thank you!"]
# Tokenize the new data
inputs = ____(____, return_tensors="pt", padding=True, truncation=True)
# Pass the tokenized inputs through the model
with ____:
outputs = ____
# Extract the new predictions
predicted_labels = ____.____(outputs.logits, dim=1).tolist()
label_map = {0: "Low risk", 1: "High risk"}
for i, predicted_label in enumerate(predicted_labels):
churn_label = label_map[predicted_label]
print(f"\n Input Text {i + 1}: {input_text[i]}")
print(f"Predicted Label: {predicted_label}")