Skip to content

toddkramer/DocumentClassifier

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 

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

  • iOS 11.0+, macOS 10.13+, tvOS 11.0+, watchOS 4.0+
  • 100% Test Coverage
  • Best CV Score: 0.965333333333

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.

NewsClassifierExample

Model

Author

Todd Kramer, todd@tekramer.com

References

About

Swift framework for document classification using a Core ML model.

Resources

License

Stars

Watchers

Forks

Packages

No packages published