/Name-Gender-Classification

Deep Learning models to Recognise the gender of an Indian Name

Primary LanguagePython

Name Gender Classification

It is a simple project to predict the gender(Male,Female) of Indian Names. I have used various Deep Learning model to get the things done.

About Dataset

The dataset I have used is present here and here

The dataset is balanced with a total of 15382 Female and 14845 male names. There were few noises like name with address, name with son of/ daughter of etc.

Model

I have used Keras as my deep learning framework for the project.

  • First model I have tried was a Simple MLP with input as one hot encoded names as input for testing purpose. The acuracy I got on test set was 79.23%.

  • The second model I have tried was Deep Character level LSTM with 30,70 and 30 units. The accuracy I got was 91.69%. The Classification Report is

Precision Recall F1 Score
Female 0.92 0.92 0.92
Male 0.91 0.92 0.91