Downloads · 30 days
18
2% of all-time downloads
finetuning-sentiment-classification-model-with-amazon-appliances-data
m-aamir95/finetuning-sentiment-classification-model-with-amazon-appliances-data
finetuning-sentiment-classification-model-with-amazon-appliances-data is a text classification model from m-aamir95. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as apache-2.0.
<h2About the model</h2 This model has been fine-tuned as part of learning the HuggingFace trainer API for the transformers.
All-time downloads
722
Public
Parameters
67M
536 MB on disk
Likes
1
Public
Repo makeup
Click a slice to open those files.
.safetensors268 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- apache-2.0
- Model type
- distilbert
- Access
- Public
- Created
- Nov 3, 2023
- Updated
- Nov 8, 2023
- SHA
- 49ba01c6
Try a prompt
Model card
From the Hugging Face model README
<h2>About the model</h2>
This model has been fine-tuned as part of learning the HuggingFace trainer API for the transformers.
This model has been fine-tuned over a sub-sample of the Amazon product reviews dataset, specifically the Amazon appliance dataset. The fine-tuned model is able to pick and classify the sentiment of product reviews successfully. <i>However, because this model has been fine-tuned over a sample of products, therefore, it is recommended to use this model with caution.</i>
<h2>About the dataset</h2> A subsample of the <a href="https://cseweb.ucsd.edu/~jmcauley/datasets/amazon_v2/">Amazon appliance reviews dataset</a> has been used to fine-tune the original Distilbert model. <h2>Fine-tune results</h2> <table> <tr> <th>Metric</th> <th>Value</th> </tr> <tr> <td>Accuracy</td> <td>0.94</td> </tr> <tr> <td>F1 Score</td> <td>0.96</td> </tr> </table>- Task
- Text Classification
- Library
- transformers
- Type
- distilbert
- License
- apache-2.0
- Languages
- en
- Created
- Nov 3, 2023
- Updated
- Nov 8, 2023