Packages

simple_bayes

1.0.0
1.0.0 0.12.2 0.12.1 0.12.0 0.11.0 0.10.0 0.9.0 0.8.0 0.7.1 0.7.0 0.6.0 0.5.2 0.5.1 0.5.0 0.4.4 0.4.3 0.4.2 0.4.1 0.4.0 0.3.1 0.3.0 0.2.0 0.1.1 0.1.0

A Simple Bayes (a.k.a. Naive Bayes) implementation in Elixir.

HexDocs

Current section

Files

Jump to
Readme
24 Versions
2 Dependencies
Dependants
Files
Activity
simple_bayes lib simple_bayes classifier.ex
Raw

lib/simple_bayes/classifier.ex 

defmodule SimpleBayes.Classifier do

  alias SimpleBayes.{Classifier.Probability, Tokenizer, Trainer.TokenStemmer}



  def classify_one(pid, string, opts) do

    classify(pid, string, opts) |> Enum.at(0) |> Kernel.elem(0)

  end



  def classify(pid, string, opts) do

    data = Agent.get(pid, &(&1))

    opts = Keyword.merge(data.opts, opts)



    data

    |> Probability.for_collection(opts[:model], category_map(string, opts))

    |> Enum.sort(&(Kernel.elem(&1,1) > Kernel.elem(&2,1)))

    |> take_top(opts[:top])

  end



  defp category_map(string, opts) do

    string

    |> Tokenizer.tokenize()

    |> TokenStemmer.stem(opts[:stem])

    |> Tokenizer.map_values(opts[:smoothing])

  end



  defp take_top(result, nil), do: result

  defp take_top(result, num) when Kernel.is_integer(num) do

    result

    |> Enum.chunk(num)

    |> Enum.at(0)

  end

end