Fine-tuning para classificação de avaliações
Seu projeto continua e agora você quer classificar as avaliações como positivas, neutras ou negativas. Você usará um modelo com fine-tuning para categorizar as avaliações automaticamente.
As classes TrainingArguments e Trainer foram importadas de transformers.
Os objetos training_args, model e tokenized_datasets também já foram pré-importados.
Este exercício faz parte do curso
Reinforcement Learning from Human Feedback (RLHF)
Instruções do exercício
- Crie os conjuntos de treino e teste usando os subconjuntos
"train"e"test"do conjunto de dadostokenized_datasets. - Inicialize a classe do trainer.
- Adicione os argumentos de treino à classe.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
# 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=____
)