Class: JrubyMahout::Evaluator

Inherits:
Object
  • Object
show all
Defined in:
lib/jruby_mahout/evaluator.rb

Instance Method Summary collapse

Constructor Details

#initialize(data_model, recommender_builder) ⇒ Evaluator

Returns a new instance of Evaluator.



5
6
7
8
9
 
# File 'lib/jruby_mahout/evaluator.rb', line 5

def initialize(data_model, recommender_builder)
  @data_model = data_model
  @recommender_builder = recommender_builder
  @mahout_evaluator = AverageAbsoluteDifferenceRecommenderEvaluator.new()
end

Instance Method Details

#evaluate(training_percentage, evaluation_percentage) ⇒ Object 



11
12
13
14
15
16
17
 
# File 'lib/jruby_mahout/evaluator.rb', line 11

def evaluate(training_percentage, evaluation_percentage)
  if @recommender_builder.recommender_name == "GenericItemBasedRecommender" and !@recommender_builder.item_based_allowed
    nil
  else
    Float(@mahout_evaluator.evaluate(@recommender_builder, nil, @data_model, training_percentage, evaluation_percentage))
  end
end