πŸ€– A suite of tools and examples for training Core ML models with Create ML.
Switch branches/tags
Nothing to show
Clone or download
pdil Remove beta software note
macOS 10.14, Xcode 10, and Swift 4.2 are no longer in beta.
Latest commit 44c10e3 Oct 22, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
Models Fix decision tree classifier playground Jul 14, 2018
Playgrounds Fix metadata Jul 14, 2018
.gitignore Add .gitignore Jun 9, 2018
LICENSE Initial commit Jun 9, 2018
README.md Remove beta software note Oct 22, 2018

README.md

ClassifierKit

πŸ€– A suite of tools and examples for training Core ML models with Create ML.

πŸ“„ Requirements

  • macOS 10.14 (Mojave) or later (download)
  • Xcode 10 or later (download)
  • Swift 4.2 or later

Important Note: Create ML is not available for the iOS SDK. It can only be used on macOS to train models and is not intended for on-device training. Instead, it is used to train models with data (which may take minutes to hours depending on computing power, size of dataset, and model). When the model is trained, a .mlmodel file can be exported and implemented in iOS/tvOS/watchOS/macOS apps using Core ML.

βš™οΈ Models

The following models are available as example Playgrounds:

Model Associated Type Playground
πŸŒ… Image Classifier MLImageClassifier
πŸŒ… Image Classifier Builder MLImageClassifierBuilder πŸ”—
πŸ“„ Text Classifier MLTextClassifier πŸ”—
🏷️ Word Tagger MLWordTagger πŸ”—
πŸ“Š Decision Tree Classifier MLDecisionTreeClassifier πŸ”—
πŸ“Š Random Forest Classifier MLRandomForestClassifier πŸ”—
πŸ“Š Boosted Tree Classifier MLBoostedTreeClassifier
πŸ“Š Logistic Regression Classifier MLLogisticRegressionClassifier
πŸ“Š Support Vector Classifier MLSupportVectorClassifier
πŸ“ˆ Linear Regressor MLLinearRegressor
πŸ“ˆ Decision Tree Regressor MLDecisionTreeRegressor
πŸ“ˆ Boosted Tree Regressor MLBoostedTreeRegressor
πŸ“ˆ Random Forest Regressor MLRandomForestRegressor

Note: Some of these are incomplete and are currently being added. The goal is to eventually have comprehensive example playgrounds for each model type in Create ML, including sample data and explanations. See Project #1 to track the progress of these playgrounds.

πŸ“ Usage

The easiest way to begin using ClassifierKit is to clone it directly onto your computer.

  1. Navigate to the desired directory on your local filesystem.
$ cd Desktop/or/any/other/folder
  1. Clone this repository:
$ git clone https://github.com/pdil/ClassifierKit.git
  1. Begin! The Playgrounds folder contains Swift Playgrounds for the many models contained within Create ML that will allow you to set parameters and begin training the models, either with the provided sample data or your own data.

πŸ—ƒοΈ References

Datasets