CommencerCommencer gratuitement

Ajustement fin pour la classification des avis

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

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

Cet exercice fait partie du cours

Reinforcement Learning from Human Feedback (RLHF)

Afficher le cours

Instructions

  • Créez les jeux de données 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 trainer.
  • Ajoutez les paramètres d’entraînement à la classe.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# 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