Adapt logistic regression to DataSet