part_5

Part 5: Finetuning for Classification

 

Main Code

• 01_main-code contains the main part code

 

Bonus Materials

• 02_bonus_additional-experiments includes additional experiments (e.g., training the last vs first token, extending the input length, etc.)
• 03_bonus_imdb-classification compares the LLM from part 6 with other models on a 50k IMDB movie review sentiment classification dataset
• 04_user_interface implements an interactive user interface to interact with the pretrained LLM