ComeçarComece de graça

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)

Ver curso

Instruções do exercício

  • Crie os conjuntos de treino e teste usando os subconjuntos "train" e "test" do conjunto de dados tokenized_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=____
)
Editar e executar o código