De voorkeursgegevensset voorbereiden
In deze oefening werk je met een gegevensset met menselijke feedback in de vorm van "chosen"- en "rejected"-uitvoer. Je taak is om de prompts uit de kolom "chosen" te halen en de data voor te bereiden voor het trainen van een rewardmodel.
De functie load_dataset uit datasets is al voor je geïmporteerd
Deze oefening maakt deel uit van de cursus
Reinforcement Learning from Human Feedback (RLHF)
Oefeninstructies
- Laad de
qgallouedec/hh-rlhf-helpful-base-trl-style-gegevensset van Hugging Face. - Schrijf een functie die de prompt uit het veld
'content'haalt, ervan uitgaande dat de prompt te vinden is op index0van de invoer van de functie. - Pas de functie die de prompt extraheert toe op de subset
'chosen'van de gegevensset.
Interactieve oefening met praktijkervaring
Probeer deze oefening door deze voorbeeldcode aan te vullen.
# Load the dataset
preference_data = ____
# Define a function to extract the prompt
def extract_prompt(text):
____
return prompt
# Apply the function to the dataset
preference_data_with_prompt = ____(
lambda sample: {**sample, 'prompt': ____(sample['chosen'])}
)
sample = preference_data_with_prompt.select(range(1))
print(sample['prompt'])