ComenzarEmpieza gratis

Utilizando el modelo afinado

El modelo se ha afinado. Ahora estás listo para utilizarlos en algunos datos nuevos y generar algunas clasificaciones. Veamos lo bien que funciona tu modelo afinado a la hora de etiquetar las nuevas interacciones como de bajo o alto riesgo de abandono.

Tu model afinado y el tokenizer se han cargado para ti.

Este ejercicio forma parte del curso

Introduction to LLMs in Python

Ver curso

Instrucciones de ejercicio

  • Tokeniza los nuevos datos.
  • Pasa las entradas tokenizadas al modelo afinado, desactivando los gradientes.
  • Extrae las nuevas predicciones.

Ejercicio interactivo práctico

Pruebe este ejercicio completando este código de muestra.

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}")
Editar y ejecutar código