Jezia's picture
Update README.md
2717c7c
|
raw
history blame
743 Bytes
metadata
language: en
tags:
  - gpt
  - text-generation
license: gpl
widget:
  - text: 'Once upon a time, '

Keras Implementation of Text generation with a miniature GPT

This repo contains the model and the notebook to this Keras example on Text generation with a miniature GPT.

Full credits to: fchollet

Background Information

This example demonstrates how to implement text generation with a miniature GPT model. The model consists of a single Transformer block with causal masking in its attention layer.

Datasets

IMDB sentiment classification dataset for training. The model generates new movie reviews for a given prompt