@@ -31,7 +31,7 @@ defmodule LearnKit.NaiveBayes.Gaussian do
3131 %LearnKit.NaiveBayes.Gaussian{data_set: [], fit_data: []}
3232
3333 """
34- @ spec new ( ) :: % LearnKit.NaiveBayes. Gaussian{ data_set: [ ] }
34+ @ spec new ( ) :: % Gaussian { data_set: [ ] }
3535
3636 def new do
3737 [ ]
@@ -51,7 +51,7 @@ defmodule LearnKit.NaiveBayes.Gaussian do
5151 %LearnKit.NaiveBayes.Gaussian{data_set: [a1: [[1, 2], [2, 3]], b1: [[-1, -2]]], fit_data: []}
5252
5353 """
54- @ spec new ( data_set ) :: % LearnKit.NaiveBayes. Gaussian{ data_set: data_set }
54+ @ spec new ( data_set ) :: % Gaussian { data_set: data_set }
5555
5656 def new ( data_set ) do
5757 % Gaussian { data_set: data_set }
@@ -67,11 +67,11 @@ defmodule LearnKit.NaiveBayes.Gaussian do
6767
6868 ## Examples
6969
70- iex> classificator |> LearnKit.NaiveBayes.Gaussian.add_train_data({:a1, [-1, -1]})
70+ iex> classificator = classificator |> LearnKit.NaiveBayes.Gaussian.add_train_data({:a1, [-1, -1]})
7171 %LearnKit.NaiveBayes.Gaussian{data_set: [a1: [[-1, -1]]], fit_data: []}
7272
7373 """
74- @ spec add_train_data ( % LearnKit.NaiveBayes. Gaussian{ data_set: data_set } , point ) :: % LearnKit.NaiveBayes. Gaussian{ data_set: data_set }
74+ @ spec add_train_data ( % Gaussian { data_set: data_set } , point ) :: % Gaussian { data_set: data_set }
7575
7676 def add_train_data ( % Gaussian { data_set: data_set } , { key , value } ) do
7777 features = if Keyword . has_key? ( data_set , key ) , do: Keyword . get ( data_set , key ) , else: [ ]
@@ -88,7 +88,7 @@ defmodule LearnKit.NaiveBayes.Gaussian do
8888
8989 ## Examples
9090
91- iex> classificator |> LearnKit.NaiveBayes.Gaussian.fit
91+ iex> classificator = classificator |> LearnKit.NaiveBayes.Gaussian.fit
9292 %LearnKit.NaiveBayes.Gaussian{
9393 data_set: [a1: [[-1, -1]]],
9494 fit_data: [
@@ -100,7 +100,7 @@ defmodule LearnKit.NaiveBayes.Gaussian do
100100 }
101101
102102 """
103- @ spec fit ( % LearnKit.NaiveBayes. Gaussian{ data_set: data_set } ) :: % LearnKit.NaiveBayes. Gaussian{ data_set: data_set , fit_data: fit_data }
103+ @ spec fit ( % Gaussian { data_set: data_set } ) :: % Gaussian { data_set: data_set , fit_data: fit_data }
104104
105105 def fit ( % Gaussian { data_set: data_set } ) do
106106 % Gaussian { data_set: data_set , fit_data: fit_data ( data_set ) }
@@ -119,7 +119,7 @@ defmodule LearnKit.NaiveBayes.Gaussian do
119119 {:ok, [a1: 0.0359, a2: 0.0039]}
120120
121121 """
122- @ spec predict_proba ( % LearnKit.NaiveBayes. Gaussian{ fit_data: fit_data } , feature ) :: { :ok , predictions }
122+ @ spec predict_proba ( % Gaussian { fit_data: fit_data } , feature ) :: { :ok , predictions }
123123
124124 def predict_proba ( % Gaussian { fit_data: fit_data } , feature ) do
125125 result = fit_data |> classify_data ( feature )
@@ -139,7 +139,7 @@ defmodule LearnKit.NaiveBayes.Gaussian do
139139 {:ok, {:a1, 0.334545454}}
140140
141141 """
142- @ spec predict ( % LearnKit.NaiveBayes. Gaussian{ fit_data: fit_data } , feature ) :: { :ok , prediction }
142+ @ spec predict ( % Gaussian { fit_data: fit_data } , feature ) :: { :ok , prediction }
143143
144144 def predict ( % Gaussian { fit_data: fit_data } , feature ) do
145145 result = fit_data |> classify_data ( feature ) |> Enum . sort_by ( & ( elem ( & 1 , 1 ) ) ) |> Enum . at ( - 1 )
@@ -159,7 +159,7 @@ defmodule LearnKit.NaiveBayes.Gaussian do
159159 {:ok, 0.857143}
160160
161161 """
162- @ spec score ( % LearnKit.NaiveBayes. Gaussian{ data_set: data_set , fit_data: fit_data } ) :: { :ok , number }
162+ @ spec score ( % Gaussian { data_set: data_set , fit_data: fit_data } ) :: { :ok , number }
163163
164164 def score ( % Gaussian { data_set: data_set , fit_data: fit_data } ) do
165165 result = fit_data |> calc_score ( data_set )
0 commit comments