Class: TLearn::RunTLearn
- Inherits:
-
Object
- Object
- TLearn::RunTLearn
- Defined in:
- lib/tlearn/run_tlearn.rb
Defined Under Namespace
Classes: UntrainedError
Instance Method Summary collapse
- #fitness(data, number_of_sweeps = @config.number_of_sweeps) ⇒ Object
-
#initialize(config = {}, working_dir = nil) ⇒ RunTLearn
constructor
A new instance of RunTLearn.
- #train(training_data, number_of_sweeps = @config.number_of_sweeps) ⇒ Object
Constructor Details
Instance Method Details
#fitness(data, number_of_sweeps = @config.number_of_sweeps) ⇒ Object
11 12 13 14 15 16 17 18 19 |
# File 'lib/tlearn/run_tlearn.rb', line 11 def fitness(data, number_of_sweeps = @config.number_of_sweeps) raise UntrainedError.new("Train me first!") unless network_trained? clear_previous_fitness_session @config.setup_fitness_data(data) execute_tlearn_fitness(number_of_sweeps) end |
#train(training_data, number_of_sweeps = @config.number_of_sweeps) ⇒ Object
21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
# File 'lib/tlearn/run_tlearn.rb', line 21 def train(training_data, number_of_sweeps = @config.number_of_sweeps) clear_entire_training_data @config.setup_config(training_data) @config.setup_training_data(training_data) execute_tlearn_train(number_of_sweeps) if training_successful?(number_of_sweeps) weights = File.read("#{@config.working_dir}/#{Config::TLEARN_NAMESPACE}.#{number_of_sweeps}.wts").split("\n") `cp #{@config.working_dir}/#{Config::TLEARN_NAMESPACE}.#{number_of_sweeps}.wts #{@config.working_dir}/#{Config::TLEARN_NAMESPACE}.wts` weights.map{|line| line.split("\t").map{|number| number.strip}} else false end end |