Downloads · 30 days
0
lora_pandas_model
samsonleegh/lora_pandas_model
lora_pandas_model is a machine learning model from samsonleegh. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
- Finetuned to generate pandas codes given a dataframe and user query. - ~100 datasets were taken from kaggle https://www.kaggle.com/datasets?search=Tabular+data - These dataset were used to generate 390 sets of data…
Access
Public
Updated Jul 25, 2024
Repo size
168 MB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors168 MB · 95%
At a glance
- Library
- transformers
- License
- apache-2.0
- Access
- Public
- Created
- Jul 25, 2024
- Updated
- Jul 25, 2024
- SHA
- 5fe4d751
Base models
Model card
From the Hugging Face model README
Uploaded model
- Finetuned to generate pandas codes given a dataframe and user query.
- ~100 datasets were taken from kaggle https://www.kaggle.com/datasets?search=Tabular+data
- These dataset were used to generate 390 sets of data queries and pandas code answers via llama3-70b https://www.kaggle.com/code/samsonleegh/sampling-data-qns-and-pandas-ans-from-dataset
- Finetuned llama3-8b-4bit with LoRA 16 adapters on 350 queries and answers pair https://colab.research.google.com/drive/1UkqjHIq-mP22AfHZCWz4kiU7hcWaXfgi?usp=sharing
- Compare ROUGE score of original vs finetuned model on 40 queries and answers pair
ROUGE Score Comparison
| Metric | llama3-8b | llama3-8b finetuned |
|---|---|---|
| ROUGE-1 | 0.4415 | 0.6585 |
| ROUGE-2 | 0.2480 | 0.4810 |
| ROUGE-L | 0.3155 | 0.5552 |
| ROUGE-Lsum | 0.3013 | 0.5570 |
- Developed by: samsonleegh
- License: apache-2.0
- Finetuned from model : unsloth/llama-3-8b-bnb-4bit
This llama model was trained 2x faster with Unsloth and Huggingface's TRL library.
- Library
- transformers
- License
- apache-2.0
- Languages
- en
- Created
- Jul 25, 2024
- Updated
- Jul 25, 2024