class TensorStream::EvaluatorUtils

Public Class Methods

get_evaluator_classes(evaluators) click to toggle source
# File lib/tensor_stream/evaluator/evaluator_utils.rb, line 5
def self.get_evaluator_classes(evaluators)
  @evaluator_classes ||= if evaluators.is_a?(Array)
    if evaluators.empty?
      TensorStream::Evaluator.default_evaluators
    else
      evaluators.collect { |name| Object.const_get("TensorStream::Evaluator::#{camelize(name.to_s)}") }
    end
  elsif evaluators.nil?
    TensorStream::Evaluator.default_evaluators
  else
    [Object.const_get("TensorStream::Evaluator::#{camelize(evaluators.to_s)}")]
  end
  @evaluator_classes
end