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)
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éestokenized_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=____
)