Downloads · 30 days
10
32% of all-time downloads
gpt2-imdb-0.1pos-0.9neg-morl
Samzy17/gpt2-imdb-0.1pos-0.9neg-morl
gpt2-imdb-0.1pos-0.9neg-morl is a machine learning model from Samzy17. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
lvwerra/gpt2-imdb finetuned using Multi-Objective Reinforcement Learning (MORL). Using positive and negative movie review classifier (sentiment-analysis) we try to optimize the reward: 0.1 positive score + 0.9 negativ…
All-time downloads
31
Public
Parameters
124M
498 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors498 MB · 99%
At a glance
- Model type
- gpt2
- Access
- Public
- Created
- Mar 15, 2025
- Updated
- Mar 17, 2025
- SHA
- 2449c2f3
Model card
From the Hugging Face model README
Description
lvwerra/gpt2-imdb finetuned using Multi-Objective Reinforcement Learning (MORL). Using positive and negative movie review classifier (sentiment-analysis) we try to optimize the reward: 0.1 * positive score + 0.9 * negative score.
Other scalar combinations were tried as well. Results are below:
Usage
Use the model.generate() function to generate reviews that, in this case, should reflect a pretty negative sentiment.
- Type
- gpt2
- Created
- Mar 15, 2025
- Updated
- Mar 17, 2025