restore Rakefile with model training task