Class: FeatureSelection::ChiSquared

Inherits:
Base
  • Object
show all
Defined in:
lib/feature_selection/chi_squared.rb

Instance Method Summary collapse

Methods inherited from Base

#classes, #initialize

Constructor Details

This class inherits a constructor from FeatureSelection::Base

Instance Method Details

#rank_features ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/feature_selection/chi_squared.rb', line 8

def rank_features
  # Returns:
  #=> {:class => {'term' => score, 'term' => score}}
  @results = {}
  
  classes.each do |klass|
    @results[klass] = {}
    
    uniq_terms.each do |term|
      answer = calculate_contribution(term, klass)
      @results[klass][term] = answer
    end #terms.each
  end #classes.each
  @results
end