SwiftNaiveBayes1.0.2

 TestsTested ✓ LangLanguage SwiftSwift License MIT ReleasedLast Release Nov 2016 SwiftSwift Version 3.0 SPMSupports SPM ✓

Maintained by akimach.

 Total 42 Week 0 Month 2

Installs

 Apps 8 Test Targets 1

GitHub

 Stars 3 Watchers 2 Forks 1 Issues 0 Contributors 1 Pull Requests 0

Code

 Files 1 LOCLines of Code 301

SwiftNaiveBayes1.0.2

• By
• akimach

Naive Bayes classifier implemented in Swift

Description

Implementation of Naive Bayes Classifier algorithm in Swift3.x, supporting Gaussian naive Bayes, Multinomial naive Bayes and Bernoulli naive Bayes

Usage

SwiftNaiveBayes is same as Scikit learn’s interface.

```import SwiftNaiveBayes

let nb = NaiveBayes()
// Positive tokens and the frequencies ["token A": Frequency of token A, ...]
let pos = ["computer": 3, "programming": 2, "python": 1, "swift": 2]
// Negative tokens ["token X": Frequency of token X, ...]
let neg = ["game": 2, "computer": 2, "video": 1, "programming": 1]
// Positive tokens for testing
let posTest = ["computer": 2, "ruby": 1, "swift": 1, "programming": 1]
// Train model
// ["Label A": ["token A": Frequency of token A, ...]]
nb.fit(["positive": pos, "negative": neg])
// Predicts log probabilities for each label
let logProbs = nb.predict(posTest)
print(logProbs) //=> ["positive": -8.9186205290602363, "negative": -10.227308671603783]
// Use method chaining
nb.fit(["positive": pos, "negative": neg]).predict(posTest)

// Save session
try! nb.save("nb.session")
// Restore session
let nb2 = NaiveBayes("nb.session")```

Install

Swift Package Manager

`Package.swift`:

```import PackageDescription

let package = Package(
name: "MyApp",
targets: [],
dependencies: [
.Package(url: "https://github.com/akimach/SwiftNaiveBayes.git", majorVersion: 1),
]
)```

MIT

akimach