panther97 / name-generator

LSTM model trained on a small dataset of 3000 names. Model generates names from model by selecting one out of top 3 letters suggested by model at a time until an EOS (End Of Sentence) character is not encountered.

Name-Generator

The project is an LSTM model written in PyTorch and trained on a small dataset of 3000 names (names.txt). Model generates names from model by selecting one out of top 3 letters suggested by model at a time until an EOS (End Of Sentence) character is not encountered.
Input for the model is any letter than you want the suggested names to start from.
Output are the 20 names generated by model.

Epoch vs Loss Graph

20 names generated by model for input of letter 'a' and 'x'