divamgupta / lstm-gender-predictor

Predict the gender of a name using LSTM

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Simple LSTM based Gender Predictor

Predict the gender of a name using LSTM

Requirements:

  • keras
  • h5py

How to use :

To train the model:

THEANO_FLAGS=device=gpu,floatX=float32 python train_genders.py

Command to see test the trained results:

python genderAPI.py <male_textfile> <female_textfile>

About

Predict the gender of a name using LSTM


Languages

Language:Python 100.0%