finetuning llama-2 7b model based on UCI drug recommendation dataset to identify the disease/condition through the feature fields (Drug, rating, reviews)
This project contains two notebooks
- The finetuning cdde for finetuning llama-2
- Inferencing using the finetuning code
These notebooks are platform-dependent and the platform used for the project is google colab a 14 GB T4 GPU. The location of the files are specific to drive location mounted to the notebook and the data used in attached to the data folder.