Simple Bayes v0.7.1 SimpleBayes.Classifier.Probability

Summary

Functions

Calculates the probabilities for the categories based on the training set

Functions

for_collection(data, model, categories_map)

Calculates the probabilities for the categories based on the training set.

Examples

iex> SimpleBayes.Classifier.Probability.for_collection(
iex>   %SimpleBayes{
iex>     categories: %{
iex>       cat: [trainings: 1, tokens: %{"nice" => 1, "cute" => 1, "cat" => 1}],
iex>       dog: [trainings: 3, tokens: %{"nice" => 2, "dog" => 3, "cute" => 3}]
iex>     },
iex>     trainings: 4,
iex>     tokens: %{"nice" => 3, "cute" => 4, "cat" => 1, "dog" => 3},
iex>     tokens_per_training: %{
iex>       {:cat, %{"nice" => 1, "cute" => 1, "cat" => 1}} => nil,
iex>       {:dog, %{"nice" => 2, "dog" => 2}} => nil,
iex>       {:dog, %{"cute" => 1, "dog" => 1}} => nil,
iex>       {:dog, %{"cute" => 2}} => nil
iex>     }
iex>   },
iex>   :multinomial,
iex>   %{"cute" => 4, "good" => 0}
iex> )
%{cat: 0.013944237739606595, dog: 0.10054155931755744}