Surnamerator is a surname generator. It's a testbed for different network architectures inspired by Andrej Karpathy's zero-to-hero youtube series.
Live demo at the surnamerator huggingface space!
To run existing models run app.ipynb
in a jupyter notebook IDE.
To train a new GPT model, run lit_surnames.ipynb
.
app.ipynb
- example code for the huggingface gradiolit_surnames.ipynb
- Most up to date training file for the transformer architecture.surnamerator.py
- Library for network models and utils.first_names.txt
/surnames.txt
- Dataset.data/
- Dataset creation.prototypes/
- Network architecture prototypes.models/
- Saved weights and configs.
Dataset creation for surnames.txt
is done in ./data
. first_names.txt
was downloaded from makemore. Example surname data:
butterly
gawne
mouch
...
The raw data for surnames was found at fivethirtyeight, data.world, and cencus.gov.