Names CoreML Demo

A Demo application using CoreML framework for predicting gender from first names.

Model

This demo is based on An introduction to Machine Learning tutorial, which describes how to build a classifier able to distinguish between boy and girl names using datasets with the popularity of baby names over the years from The US Social Security Administration.

CoreML model was converted from Scikit-learn Pipeline using coremltools python package.

Requirements

• Xcode 9
• iOS 11

Installation

git clone https://github.com/cocoa-ai/NamesCoreMLDemo.git
cd NamesCoreMLDemo
open Names.xcodeproj/

Build the project and run it on a simulator or a device with iOS 11.

Conversion

cd Convert
python names.py

Author

Vadym Markov, [email protected]

Credits

• Is it a boy or a girl? An introduction to Machine Learning from http://nlpforhackers.io

References

• Apple Machine Learning
• CoreML Framework
• coremltools