TreeKit

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.3.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
• [ ] 0.4 Regression trees
• [ ] 0.5 Boosting