Skip to content

cnmoro/GenderClassifierByName

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Modelo básico para classificar nomes entre ‘masculino’ e ‘feminino’

Utilizando datasets de nomes do repositório: genderComputer

Consultar arquivo “model.py”.

Modelo treinado já fornecido (model.pkl)

Para utilizar em código externo, será necessário definir a classe FeatureExtractor, e o método extract_features

  • Carregar o modelo
    • model = pickle.load(open('model.pkl', 'rb'))
  • Inferir um nome
    • model.predict(['Carlo'])[0] # saida “male”

Não utilizar com sobrenomes, o modelo foi treinado somente com os primeiros nomes de indivíduos

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages