https://github.com/akimach/swiftnaivebayes

Naive Bayes classifier implemented in Swift
https://github.com/akimach/swiftnaivebayes

carthage cocoapods machine-learning machine-learning-algorithms naive-bayes-classifier naivebayes swift swift-3 swift-language swift-library

Last synced: 5 days ago
JSON representation

Naive Bayes classifier implemented in Swift

• Host: GitHub
• URL: https://github.com/akimach/swiftnaivebayes
• Owner: akimach
• License: mit
• Created: 2016-11-14T13:13:56.000Z (almost 8 years ago)
• Default Branch: master
• Last Pushed: 2017-10-27T03:51:33.000Z (almost 7 years ago)
• Last Synced: 2024-09-25T20:38:54.502Z (8 days ago)
• Topics: carthage, cocoapods, machine-learning, machine-learning-algorithms, naive-bayes-classifier, naivebayes, swift, swift-3, swift-language, swift-library
• Language: Swift
• Homepage:
• Size: 60.5 KB
• Stars: 12
• Watchers: 3
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
SwiftNaiveBayes

===

![](http://img.shields.io/badge/Swift-3.0-blue.svg)

![macOS](https://img.shields.io/badge/os-macOS-green.svg?style=flat)

![Linux](https://img.shields.io/badge/os-linux-green.svg?style=flat)

![Carthage compatible](https://img.shields.io/badge/Carthage-compatible-4BC51D.svg?style=flat)

![Swift Package Manager compatible](https://img.shields.io/badge/Swift%20Package%20Manager-compatible-brightgreen.svg)

![CocoaPods compatible](https://img.shields.io/badge/CocoaPods-compatible-4BC51D.svg)

![MIT License](http://img.shields.io/badge/license-MIT-blue.svg?style=flat)

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](http://scikit-learn.org/stable/modules/naive_bayes.html)'s interface.

```swift

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`:

```swift

import PackageDescription

let package = Package(

    name: "MyApp",

    targets: [],

    dependencies: [

        .Package(url: "https://github.com/akimach/SwiftNaiveBayes.git", majorVersion: 1),

    ]

)

```

### CocoaPods

`Podfile`:

```

platform :ios, '8.0'

use_frameworks!

target 'MyApp' do

    pod 'SwiftNaiveBayes'

end

```

### Carthage

`Cartfile`:

```

github "akimach/SwiftNaiveBayes"

```

## Licence

[MIT](https://github.com/akimach/SwiftNaiveBayes/blob/master/LICENSE)

## Author

[akimach](https://github.com/akimach)