Comece agoraComece grátis

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 exercicio faz parte do curso

Reinforcement Learning from Human Feedback (RLHF)

Ver curso

Instruções do exercicio

  • 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.

exercicio interativo prático

Tente este exercicio 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 Código