Kaive Bayes

Sorry about the ridiculous name, this is a very simple Naive Bayes classifier written in Kotlin.

Techniques used

In order to be competitive, Kaive Bayes leverages the following techniques:

• Laplace Smoothing for eliminating zero probabilities
• log-sum trick to prevent number underflows
• Normalization
• Stopword purging

Serialization is to be added soon along with various optimization enhancements.

Usage

 var bae = BayesClassifier<String>()

    bae.train(Input("A great game", "sports"))
    bae.train(Input("The election was over", "notSports"))

    bae.train(Input("Very clean match", "sports"))
    bae.train(Input("A clean but forgettable game", "sports"))
    bae.train(Input("It was a close election", "notSports"))

    val message1 = "A very close game"
    var map = bae.predict(message1)
    println("RESULTS OF TEXT: $message1")
    println("------------------------------------------------------------------")
    println("Is about sports " + map["sports"])
    println("Is NOT about sports " + map["notSports"])
    println()

For more examples see the Main function

Add it to your project

You may find kaive-bayes under jcenter.

Gradle

compile 'io.github.thanosfisherman:kaive-bayes:<latest_version_here>'

Maven

<dependency>
  <groupId>io.github.thanosfisherman</groupId>
  <artifactId>kaive-bayes</artifactId>
  <version>latest_version_here</version>
  <type>pom</type>
</dependency>

Contributing?

Feel free to add/correct/fix something to this library, I will be glad to improve it with your help.

License

Copyright 2018 Thanos Psaridis

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

   http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.