Home

Awesome

Names CoreML Demo

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

<div align="center"> <img src="https://github.com/cocoa-ai/NamesCoreMLDemo/blob/master/Screenshot.png" alt="NamesCoreMLDemo" width="300" height="533" /> </div>

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

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, markov.vadym@gmail.com

Credits

References