Home

Awesome

DocumentClassifier

CocoaPods Version Swift Platform

Overview

DocumentClassifier is a Swift framework for classifying documents into one of five categories (Business, Entertainment, Politics, Sports, and Technology). It uses a CoreML model trained with 1,500 news articles from the BBC.

Features

Usage

Swift 4 Released (Sample Article)

let text = articleText
guard let classification = classifier.classify(text) else { return }
print(classification.prediction) // Technology: 0.42115752953489294
print(classification.allResults) // Business: 0.141, Entertainment: 0.138, Politics: 0.113, Sports: 0.187, Technology: 0.421

Installation

CocoaPods

CocoaPods is a centralized dependency manager for Cocoa projects. To install DocumentClassifier with CocoaPods:

  1. Make sure the latest version of CocoaPods is installed.

  2. Add DocumentClassifier to your Podfile:

use_frameworks!

pod 'DocumentClassifier', '1.2.0'
  1. Run pod install.

Example App

NewsClassifier is an example app using the framework.

<div align="center"> <img src="https://github.com/toddkramer/DocumentClassifier/blob/master/NewsClassifier/Screenshot.png" alt="NewsClassifierExample" width="310" height="382" /> </div>

Model

Author

Todd Kramer, todd@tekramer.com

References