module FisherClassifier

Constants

VERSION

Public Class Methods

create(&block) click to toggle source
# File lib/fisher_classifier.rb, line 9
def create(&block)
  config = Config.new block

  Classifier.new config
end
create_in_memory() click to toggle source
# File lib/fisher_classifier.rb, line 15
def create_in_memory
  create do
    @features = {}
    @categories = {}

    inc_feature do |feature, category|
      @features[category] ||= {}
      @features[category][feature] ||= 0
      @features[category][feature] += 1
    end

    inc_category do |category|
      @categories[category] ||= 0
      @categories[category] += 1
    end

    get_features do |text|
      text.split(' ').map { |s| s.downcase }
    end

    categories do
      [:good, :bad]
    end

    category_count do |category|
      if @categories.has_key?(category)
        @categories[category]
      else
        0
      end
    end

    features_count do |feature, category|
      if @features.has_key?(category) && @features[category].has_key?(feature)
        @features[category][feature]
      else
        0
      end
    end

    default_category do
      :none
    end

  end

end