Examples of SGDTrainConfig

Examples of org.gd.spark.opendl.downpourSGD.SGDTrainConfig

31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
      JavaRDD<SampleVector> rdds = context.parallelize(trainList);
      rdds.count();
      logger.info("RDD ok.");
     
      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.setMrDataStorage(StorageLevel.MEMORY_ONLY());
            config.setPrintLoss(true);
            config.setLossCalStep(3);
            config.setParamOutput(true);
            config.setParamOutputStep(3);
            config.setParamOutputPath("wb.bin");
           
            logger.info("Start to train lr.");
            DownpourSGDTrain.train(lr, rdds, config);
           
            int trueCount = 0;
View Full Code Here

Examples of org.gd.spark.opendl.downpourSGD.SGDTrainConfig

23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
      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;
View Full Code Here

Examples of org.gd.spark.opendl.downpourSGD.SGDTrainConfig

24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
      List<SampleVector> trainList = new ArrayList<SampleVector>();
      List<SampleVector> testList = new ArrayList<SampleVector>();
      DataInput.splitList(samples, trainList, testList, 0.7);
     
      AutoEncoder da = new AutoEncoder(x_feature, n_hidden);
            SGDTrainConfig config = new SGDTrainConfig();
            config.setUseCG(true);
            config.setDoCorruption(false);
            config.setCorruption_level(0.25);
            config.setCgEpochStep(50);
            config.setCgTolerance(0);
            config.setCgMaxIterations(30);
            config.setMaxEpochs(50);
            config.setNbrModelReplica(4);
            config.setMinLoss(0.01);
            config.setUseRegularization(true);
            config.setPrintLoss(true);
           
            logger.info("Start to train dA.");
            DownpourSGDTrain.train(da, trainList, config);
           
            double[] reconstruct_x = new double[x_feature];
View Full Code Here

Examples of org.gd.spark.opendl.downpourSGD.SGDTrainConfig

24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
      List<SampleVector> trainList = new ArrayList<SampleVector>();
      List<SampleVector> testList = new ArrayList<SampleVector>();
      DataInput.splitList(samples, trainList, testList, 0.7);
     
      RBM rbm = new RBM(x_feature, n_hidden);
            SGDTrainConfig config = new SGDTrainConfig();
            config.setUseCG(true);
            config.setCgEpochStep(50);
            config.setCgTolerance(0);
            config.setCgMaxIterations(30);
            config.setMaxEpochs(50);
            config.setNbrModelReplica(4);
            config.setMinLoss(0.01);
            config.setPrintLoss(true);
           
            logger.info("Start to train RBM.");
            DownpourSGDTrain.train(rbm, trainList, config);
           
            double[] reconstruct_x = new double[x_feature];
View Full Code Here

Examples of org.gd.spark.opendl.downpourSGD.SGDTrainConfig

31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
     
      int[] hiddens = new int[1];
            hiddens[0] = 160;
           
      BP bp = new BP(x_feature, y_feature, hiddens);
            SGDTrainConfig config = new SGDTrainConfig();
            config.setUseCG(true);
            config.setCgEpochStep(50);
            config.setCgTolerance(0);
            config.setCgMaxIterations(30);
            config.setMaxEpochs(50);
            config.setNbrModelReplica(4);
            config.setMinLoss(0.01);
            config.setUseRegularization(true);
            config.setPrintLoss(true);
            config.setCgInitStepSize(1.0);

            logger.info("Start to train bp.");
            DownpourSGDTrain.train(bp, trainList, config);
           
//            int trueCount = 0;
View Full Code Here

Examples of org.gd.spark.opendl.downpourSGD.SGDTrainConfig

32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
      JavaRDD<SampleVector> rdds = context.parallelize(trainList);
      rdds.count();
      logger.info("RDD ok.");
     
      AutoEncoder da = new AutoEncoder(x_feature, n_hidden);
            SGDTrainConfig config = new SGDTrainConfig();
            config.setUseCG(true);
            config.setDoCorruption(true);
            config.setCorruption_level(0.25);
            config.setCgEpochStep(50);
            config.setCgTolerance(0);
            config.setCgMaxIterations(10);
            config.setMaxEpochs(50);
            config.setNbrModelReplica(4);
            config.setMinLoss(0.01);
            config.setUseRegularization(true);
            config.setMrDataStorage(StorageLevel.MEMORY_ONLY());
            config.setPrintLoss(true);
            config.setLossCalStep(3);
           
            logger.info("Start to train dA.");
            DownpourSGDTrain.train(da, rdds, config);
           
            double[] reconstruct_x = new double[x_feature];
View Full Code Here

Examples of org.gd.spark.opendl.downpourSGD.SGDTrainConfig

32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
      JavaRDD<SampleVector> rdds = context.parallelize(trainList);
      rdds.count();
      logger.info("RDD ok.");
     
      RBM rbm = new RBM(x_feature, n_hidden);
            SGDTrainConfig config = new SGDTrainConfig();
            config.setUseCG(true);
            config.setCgEpochStep(50);
            config.setCgTolerance(0);
            config.setCgMaxIterations(10);
            config.setMaxEpochs(50);
            config.setNbrModelReplica(4);
            config.setMinLoss(0.01);
            config.setMrDataStorage(StorageLevel.MEMORY_ONLY());
            config.setPrintLoss(true);
            config.setLossCalStep(3);
           
            logger.info("Start to train RBM.");
            DownpourSGDTrain.train(rbm, rdds, config);
           
            double[] reconstruct_x = new double[x_feature];
View Full Code Here
TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.