package org.gd.spark.opendl.example.standalone;
import java.util.ArrayList;
import java.util.List;
import org.apache.log4j.Logger;
import org.gd.spark.opendl.downpourSGD.SGDTrainConfig;
import org.gd.spark.opendl.downpourSGD.SampleVector;
import org.gd.spark.opendl.downpourSGD.Softmax.LR;
import org.gd.spark.opendl.downpourSGD.train.DownpourSGDTrain;
import org.gd.spark.opendl.example.ClassVerify;
import org.gd.spark.opendl.example.DataInput;
public class LRTest {
private static final Logger logger = Logger.getLogger(LRTest.class);
public static void main(String[] args) {
try {
int x_feature = 784;
int y_feature = 10;
List<SampleVector> samples = DataInput.readMnist("mnist_784_1000.txt", x_feature, y_feature);
List<SampleVector> trainList = new ArrayList<SampleVector>();
List<SampleVector> testList = new ArrayList<SampleVector>();
DataInput.splitList(samples, trainList, testList, 0.7);
LR lr = new LR(x_feature, y_feature);
SGDTrainConfig config = new SGDTrainConfig();
config.setUseCG(true);
config.setCgEpochStep(100);
config.setCgTolerance(0);
config.setCgMaxIterations(30);
config.setMaxEpochs(100);
config.setNbrModelReplica(4);
config.setMinLoss(0.01);
config.setUseRegularization(true);
config.setPrintLoss(true);
logger.info("Start to train lr.");
DownpourSGDTrain.train(lr, trainList, config);
int trueCount = 0;
int falseCount = 0;
double[] predict_y = new double[y_feature];
for(SampleVector test : testList) {
lr.predict(test.getX(), predict_y);
if(ClassVerify.classTrue(test.getY(), predict_y)) {
trueCount++;
}
else {
falseCount++;
}
}
logger.info("trueCount-" + trueCount + " falseCount-" + falseCount);
} catch(Throwable e) {
logger.error("", e);
}
}
}