Packages

simple_bayes

0.3.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
0 Dependencies
Dependants
Files
Activity
simple_bayes lib simple_bayes.ex
Raw

lib/simple_bayes.ex 

defmodule SimpleBayes do

  defstruct categories: %{}, trainings: 0, tokens: %{}, tokens_per_training: []



  def init do

    {:ok, agent} = Agent.start_link fn -> %SimpleBayes{} end



    agent

  end



  def default_weight, do: Application.get_env(:simple_bayes, :default_weight)

  def smoothing,      do: Application.get_env(:simple_bayes, :smoothing)

  def stop_words,     do: Application.get_env(:simple_bayes, :stop_words)



  defdelegate train(agent, category, string, opts \\ []), to: SimpleBayes.Trainer

  defdelegate classify(agent, string),                    to: SimpleBayes.Classifier

  defdelegate classify_one(agent, string),                to: SimpleBayes.Classifier

end