add rep in train_model() for k-fold cross validation
get learning rate data and figure