modify readme

Author: kortirso

README.md

@@ -19,19 +19,15 @@ def deps do
 end
 ```
 
-### K-Nearest Neighbours
+### K-Nearest Neighbours classification
 
 Initialize classificator with data set consists from labels and features:
 
 ```elixir
   classificator = LearnKit.Knn.new
                   |> LearnKit.Knn.add_train_data({:a1, [-1, -1]})
                   |> LearnKit.Knn.add_train_data({:a1, [-2, -1]})
-                  |> LearnKit.Knn.add_train_data({:a1, [-3, -2]})
                   |> LearnKit.Knn.add_train_data({:a2, [1, 1]})
-                  |> LearnKit.Knn.add_train_data({:a2, [2, 1]})
-                  |> LearnKit.Knn.add_train_data({:a2, [3, 2]})
-                  |> LearnKit.Knn.add_train_data({:a2, [-2, -2]})
 ```
 
 Predict label for new feature:
@@ -44,6 +40,43 @@ Predict label for new feature:
     algorithm - algorithm for calculation of distances, one of the [brute], optional, default - "brute"
     weight - method of weighted neighbors, one of the [uniform|distance], optional, default - "uniform"
 
+### Gaussian Naive Bayes classification
+
+Initialize classificator with data set consists from labels and features:
+
+```elixir
+  classificator = LearnKit.NaiveBayes.Gaussian.new
+                  |> LearnKit.NaiveBayes.Gaussian.add_train_data({:a1, [-1, -1]})
+                  |> LearnKit.NaiveBayes.Gaussian.add_train_data({:a1, [-2, -1]})
+                  |> LearnKit.NaiveBayes.Gaussian.add_train_data({:a2, [1, 1]})
+```
+
+Fit data set
+
+```elixir
+  classificator = classificator |> LearnKit.NaiveBayes.Gaussian.fit
+```
+
+Return probability estimates for the feature
+
+```elixir
+  classificator = classificator |> LearnKit.NaiveBayes.Gaussian.predict_proba([1, 2])
+```
+    feature - new feature for prediction, required
+
+Return exact prediction for the feature
+
+```elixir
+  classificator = classificator |> LearnKit.NaiveBayes.Gaussian.predict([1, 2])
+```
+    feature - new feature for prediction, required
+
+Returns the mean accuracy on the given test data and labels
+
+```elixir
+  classificator = classificator |> LearnKit.NaiveBayes.Gaussian.score
+```
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/kortirso/elixir_learn_kit.

lib/learn_kit/naive_bayes/gaussian.ex

@@ -155,8 +155,6 @@ defmodule LearnKit.NaiveBayes.Gaussian do
 
   ## Examples
 
-      iex> classificator = LearnKit.NaiveBayes.Gaussian.new([{:label1, [[-1, -1], [-2, -1], [-3, -2]]}, {:label2, [[1, 1], [2, 1], [3, 2], [-2, -2]]}])
-      iex> classificator = classificator |> LearnKit.NaiveBayes.Gaussian.fit
       iex> classificator |> LearnKit.NaiveBayes.Gaussian.score
       {:ok, 0.857143}