|
--- |
|
library_name: peft |
|
license: apache-2.0 |
|
language: |
|
- en |
|
metrics: |
|
- accuracy |
|
pipeline_tag: text-classification |
|
tags: |
|
- opensource |
|
- finetunning |
|
- llm |
|
- sentiment-analysis |
|
--- |
|
|
|
## Model Description |
|
|
|
This repository contains a fine-tuned sentiment analysis model based on the `distilbert-base-uncased` architecture, trained on the "shawhin/imdb-truncated" dataset. The model is designed for text classification tasks in the English language. |
|
|
|
## Model Performance |
|
|
|
The model's performance is evaluated based on accuracy, a common metric for text classification tasks. The specific performance metrics may vary depending on the use case and dataset. |
|
|
|
## Training Procedure |
|
|
|
### Framework Versions |
|
|
|
- PEFT 0.5.0 |
|
|
|
### Dataset |
|
|
|
The model is trained on the "shawhin/imdb-truncated" dataset, which is a truncated version of the IMDb movie review dataset. It contains labeled movie reviews with binary sentiment labels (positive or negative). |
|
|
|
## Fine-Tuning Details |
|
|
|
The model is fine-tuned using the `distilbert-base-uncased` architecture, a smaller and faster version of BERT, well-suited for various NLP tasks. |
|
|
|
## How to Use |
|
|
|
You can use this fine-tuned sentiment analysis model for various text classification tasks, including sentiment analysis, text categorization, and more. To use the model, you can easily load it with the Hugging Face Transformers library and integrate it into your Python applications. |
|
|
|
Here's an example of how to load and use the model: |
|
|
|
```python |
|
from transformers import AutoModelForSequenceClassification, AutoTokenizer |
|
|
|
# Load the fine-tuned model |
|
model = AutoModelForSequenceClassification.from_pretrained("samadpls/sentiment-analysis") |
|
|
|
# Load the tokenizer |
|
tokenizer = AutoTokenizer.from_pretrained("your-model-name") |
|
|
|
# Perform inference |
|
text = "This is a great movie!" |
|
inputs = tokenizer(text, return_tensors="pt") |
|
outputs = model(**inputs) |
|
predicted_label = outputs.logits.argmax().item() |
|
|
|
# Print the predicted sentiment label |
|
print("Predicted Sentiment: Positive" if predicted_label == 1 else "Predicted Sentiment: Negative") |
|
``` |
|
# License |
|
This model is distributed under the Apache License 2.0. For more details, see the [LICENSE](LICENSE) file. |
