logistic regression training and test set