Gender is determined from the name.

pip install nltk
import nltk
import random
from nltk import classify
from nltk import NaiveBayesClassifier as NBC

data

Prepare a list with male names (male list) and a list with female names (female list). (Romaji) (The number of samples is the same) (Only the first name out of the first and last names)

Example: malelist = ['kazuo','kenji', ...]

name.py


def feature_extraction(word):
    return {"last":word[-3:]}
#A function that retrieves only the last 3 characters of a name

maleNames=[(name, "male")for name in malelist]
femaleNames = [(name, 'female')for name in femalelist]
allNames = maleNames + femaleNames #Join the list of men and the list of women
random.shuffle(allNames) #Shuffle the contents of the list

featureData=[(feature_extraction(n),gender) for (n,gender) in allNames]
#[(Three letters behind,sex),(Three letters behind,sex),(Three letters behind,sex)・ ・ ・]In the shape of

genderIdentifier=NBC.train(featureData)

#If you want to check the accuracy, it feels like you did,
#num=7*len(featureData)//Of 10 data
#train_data=featureData[num:]
#test_data=featureData[:num]
#genderIdentifier=NBC.train(train_data)

How to save the model you created

import pickle
f = open('my_classifier.pickle', 'wb')
pickle.dump(genderIdentifier, f)
f.close()

How to load a saved model

import pickle
f = open('my_classifier.pickle', 'rb')
classifier = pickle.load(f)
f.close()

reference

Save Naive Bayes Trained Classifier in NLTK

Machine Learning Model - Gender Identifier with NLTK in less than 15 lines of code

Machine Learning Model - Gender Identifier with NLTK in less than 15 lines of code

Recommended Posts

Gender is determined from the name.
Find out the name of the method that called it from the method that is python
I want to see the file name from DataLoader
Identify the name from the flower image with keras (tensorflow)
Find the part that is 575 from Wikipedia in Python
Read the function name from the DB and execute it dynamically
Predict gender from name using Gender API and Pykakasi in Python
Get the class name where the method is defined in the decorator
Let's search from the procession
What is the activation function?
What is the Linux kernel?
Remove the frame from the image
I tweeted from the terminal!
What is the interface for ...
What is the Callback function?
The image is a slug
What is a dog? Django--Get Name and Date from URL Volume
From the AWS cloud product page, put the AWS service name in csv