Matrix input = data_set.getFirst();
Matrix labels = data_set.getSecond();
System.out.println( "Beginning LogReg Training on CovType");
LogisticRegression logRegression = new LogisticRegression( input, input.numCols(), labels.numCols());
double learningRate = 0.001;
logRegression.labels = labels;
LogisticRegressionOptimizer opt = new LogisticRegressionOptimizer( logRegression, learningRate );
CustomConjugateGradient g = new CustomConjugateGradient(opt);
//NonZeroStoppingConjugateGradient g = new NonZeroStoppingConjugateGradient(opt);
g.optimize();
Matrix predict = logRegression.predict(input);
Evaluation eval = new Evaluation();
eval.eval(labels, predict);
//log.info(eval.stats());
System.out.println( eval.stats() );