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