CommencezCommencez gratuitement

Ajustement fin pour la classification d'avis

Votre projet se poursuit et vous souhaitez maintenant classer les avis comme positifs, neutres ou négatifs. Vous utiliserez un modèle ajusté finement pour catégoriser automatiquement les avis.

Les classes TrainingArguments et Trainer ont été importées de transformers. Les objets training_args, model et tokenized_datasets ont également été préimportés.

Cette activité fait partie du cours

Reinforcement Learning from Human Feedback (RLHF)

Voir le cours

Instructions de l’exercice

  • Créez les ensembles d'entraînement et de test en utilisant les sous-ensembles "train" et "test" du jeu de données tokenized_datasets.
  • Initialisez la classe du formateur.
  • Ajoutez les paramètres d'entraînement à la classe.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Define the train and test datasets
training_dataset = ____
testing_dataset = ____

# Initialize the trainer class
trainer = ____(
# Add arguments to the class
    model=model,
    args=training_args,
    train_dataset=____,
    eval_dataset=____
)
Modifier et exécuter le code