Simple Bayes v0.7.1 SimpleBayes.Trainer.TokenCataloger

Summary

Functions

Catalogs tokens based on their training categories

Functions

catalog(arg, category, opts)

Catalogs tokens based on their training categories.

Examples

iex> SimpleBayes.Trainer.TokenCataloger.catalog(
iex>   {
iex>     ["cute", "dog"],
iex>     %SimpleBayes{categories: %{}, trainings: 0, tokens: %{}}
iex>   },
iex>   :dog,
iex>   [default_weight: 1]
iex> )
%SimpleBayes{
  categories: %{
    dog: [trainings: 1, tokens: %{"cute" => 1, "dog" => 1}]
  }
}

iex> SimpleBayes.Trainer.TokenCataloger.catalog(
iex>   {
iex>     ["cute", "cat"],
iex>     %SimpleBayes{
iex>       categories: %{
iex>         dog: [trainings: 1, tokens: %{"cute" => 1, "dog" => 1}]
iex>       }
iex>     }
iex>   },
iex>   :cat,
iex>   [default_weight: 1]
iex> )
%SimpleBayes{
  categories: %{
    dog: [trainings: 1, tokens: %{"cute" => 1, "dog" => 1}],
    cat: [trainings: 1, tokens: %{"cute" => 1, "cat" => 1}]
  }
}

iex> SimpleBayes.Trainer.TokenCataloger.catalog(
iex>   {
iex>     ["good", "cat"],
iex>     %SimpleBayes{
iex>       categories: %{
iex>         dog: [trainings: 1, tokens: %{"cute" => 1, "dog" => 1}],
iex>         cat: [trainings: 1, tokens: %{"cute" => 1, "cat" => 1}]
iex>       }
iex>     }
iex>   },
iex>   :cat,
iex>   [default_weight: 1]
iex> )
%SimpleBayes{
  categories: %{
    dog: [trainings: 1, tokens: %{"cute" => 1, "dog" => 1}],
    cat: [trainings: 2, tokens: %{"cute" => 1, "cat" => 2, "good" => 1}]
  }
}