Added function train_dataset at trainable.lua