CommencerCommencer gratuitement

Filtering datasets for evaluation

You are building a training and evaluation pipeline for your company's health care chatbot, which is used by hospitals to onboard new patients.

Your task is to create a pipeline to load the MedQuad-MedicalQnADataset to evaluate an LLM on its ability to answer medical questions. You are asked to load the dataset in the ds variable, and only include the first 500 samples of the train split of the dataset stored in dataset_name as your evaluation set.

Cet exercice fait partie du cours

Fine-Tuning with Llama 3

Afficher le cours

Instructions

  • Import necessary functions and classes from datasets.
  • Load the dataset in the ds variable.
  • Manipulate ds to include the first 500 samples of the train split of the dataset stored in dataset_name as your evaluation set.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Load neccesary imports from library
from datasets import ____, ____

# Load the training split of the dataset
ds = load_dataset(dataset_name, split=____)

# Filter for the first 500 samples of the dataset
filtered_ds = ____
Modifier et exécuter le code