Home

Awesome

Sentiment Polarity CoreML Demo

A Demo application using CoreML framework for sentiment polarity analysis.

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

Model

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

The model is based on LinearSVC classifier and is able to distinguish between positive and negative sentences with best CV score = 0.801013024602. It was trained using Epinions.com dataset with reviews of products and services. Accuracy can be improved by using TfidfVectorizer for feature extraction, but it's not supported by coremltools at the moment.

Requirements

Installation

git clone https://github.com/cocoa-ai/SentimentCoreMLDemo.git
cd SentimentCoreMLDemo
open SentimentPolarity.xcodeproj/

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

Conversion

cd Convert
python sentiment.py

Author

Vadym Markov, markov.vadym@gmail.com

Credits

References