mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2025-08-29 11:00:55 +00:00
3c3dae0967
* Add mean pooling experiment to classifier bonus experiments * formatting * add average embeddings option * pep8
Chapter 6: Finetuning for Classification
Main Chapter Code
- 01_main-chapter-code contains the main chapter 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 chapter 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