fix bin/save_model

This commit is contained in:
Christoffer Müller Madsen 2018-03-14 11:06:55 +01:00
parent 7b0e60f069
commit 153776d83e
Signed by: christoffer
GPG Key ID: 337BA5A95E686EFD

View File

@ -2,10 +2,10 @@
require 'date'
models_dir = 'models'
model_name = ARGV[0]
model_name = (ARGV[0] or "default")
model_path = File.join(models_dir, model_name)
episode_count = (File.read File.join(model_path, 'model.episodes')).to_i
episode_count = (File.read File.join(model_path, 'episodes_trained')).to_i
puts "Found model #{model_name} with episodes #{episode_count} trained!"