Class: PokemonNameGenerator::Algorithm::Factory

Inherits:
Object
  • Object
show all
Defined in:
lib/pokemon_name_generator/algorithm/factory.rb

Instance Method Summary collapse

Constructor Details

#initialize(training_data: Corpus::Names.as_letters, **options) ⇒ Factory

Returns a new instance of Factory.



4
5
6
7
 
# File 'lib/pokemon_name_generator/algorithm/factory.rb', line 4

def initialize(training_data: Corpus::Names.as_letters, **options)
  @training_data = training_data
  @options = options
end

Instance Method Details

#build_algorithm ⇒ Object 



9
10
11
12
13
14
 
# File 'lib/pokemon_name_generator/algorithm/factory.rb', line 9

def build_algorithm
  case options.fetch(:algorithm)
  when "naive" then Naive.new(training_data)
  when "markov" then Markov.new(training_data, chain_length: context)
  end
end