GPT-2 Movie Review Generator
Performance Metrics
- Training Loss: 1.0200 (after 1 epoch)
- Hardware: Google Colab GPU/TPU
Example Generation
Prompt: This movie was
Output:
one of the first I saw in theaters and I was so excited to see it again...
How to use
import keras_hub
model = keras_hub.models.CausalLM.from_preset("hf://hinttt/gpt2-imdb-finetuned")
output = model.generate("This movie was", max_length=100)
- Downloads last month
- 3