Skip to content

Latest commit

 

History

History
31 lines (22 loc) · 895 Bytes

README.md

File metadata and controls

31 lines (22 loc) · 895 Bytes
Raw

Character-based LSTM

A LSTM that can be used as a starting point to train a character-based prediction model. This example tries to classify if a first name is female or male based on the dataset in the /data folder.

Prerequisites

  • Make sure that you have a newer version of Python and PIP installed

  • PyTorch is required. Some platforms may require that you have additional packages installed before you can run PyTorch (see the install pages)

  • Open up a terminal and type

    git clone https://github.com/christian-vorhemus/character-lstm.git
cd character-lstm
pip install -r requirements.txt

Model training

Start the training with

python main.py train

Inference

Check if a name is predicted as male or female with

python main.py anna
anna is probably a female name