Swiftpack.co - Package - PeetV/TreeKit

TreeKit

Build Status codecov Codacy Badge

TreeKit is a Swift package for classification problems, primarily drawing on decision tree based machine learning algorithms. It builds on DataKit.

Installation

Install using the Swift package manager by adding a dependency in a project Package.swift file:

...
dependencies: [
    // Dependencies declare other packages that this package depends on.
    // .package(url: /* package url */, from: "1.0.0"),
    .package(url: "https://github.com/PeetV/DataKit.git", from: "0.3.0"),
    .package(url: "https://github.com/PeetV/TreeKit.git", from: "0.4.0"),
],
targets: [
    // Targets are the basic building blocks of a package. A target can define a
    // module or a test suite.
    // Targets can depend on other targets in this package, and on products in
    // packages which this package depends on.
    .target(
        name: ...,
        dependencies: ["DataKit", "TreeKit"]),
...

Examples

See the docs folder for interactive examples in playgrounds that can be run using Xcode or see the Contents.swift file in the playground folder.

| Playground | Description | |--------------------------|---------------------------------------------------------------------------| | Overview.playground | General overview of the TreeKit API. | | ClassifierZoo.playground | Examples of applying TreeKit classification models to different datasets. |

Roadmap

  • [x] 0.1 Single binary classification tree
  • [x] 0.2 Ensemble of classification trees
  • [x] 0.3 Naive Bayes classifier
  • [x] 0.4 K-Nearest Neighbor classifier
  • [ ] 0.5 Regression trees
  • [ ] 0.6 Boosting

Github

link
Stars: 0
Help us keep the lights on

Used By

Total: 1

Releases

0.4.0 - Mar 2, 2019

K-Nearest Neighbour classifier added

0.3.0 - Feb 16, 2019

Added NaiveBayes classifier class.

0.2.0 - Feb 3, 2019

Ensemble of classification trees

0.1.0 - Jan 28, 2019

See the .playground for an API overview.