LearnKit
Installation
Add this line to your application's Gemfile:
gem 'learn_kit'And then execute:
$ bundle
Or install it yourself as:
$ gem install learn_kit
K-Nearest Neighbors
Initialize classificator with data set consists from labels and features:
data_set = { label1: [[-1, -1], [-2, -1], [-3, -2]], label2: [[1, 1], [2, 1], [3, 2], [-2, -2]] }
clf = LearnKit::Knn.new(data_set: data_set)Predict label for new feature:
clf.predict(k: 3, algorithm: 'brute', weight: 'uniform', point: [-1, -2])k - number of nearest neighbors
algorithm - algorithm for calculation of distances, one of the [brute]
weight - method of weighted neighbors, one of the [uniform|distance]
point - new feature for prediction
Naive Bayes
Gaussian
Initialize classificator with data set consists from labels and features:
data_set = { label1: [[-1, -1], [-2, -1], [-3, -2]], label2: [[1, 1], [2, 1], [3, 2], [-2, -2]] }
clf = LearnKit::NaiveBayes::Gaussian.new(data_set: data_set)Make fit of test data:
clf.fitPredict label for new feature:
clf.predict([-1, -2])Or show probability for all labels:
clf.predict_proba([-1, -2])Calculate accuracy for test data:
clf.score