Package org.gd.spark.opendl.downpourSGD.Backpropagation

Examples of org.gd.spark.opendl.downpourSGD.Backpropagation.AutoEncoder


     
      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];
            double totalError = 0;
            for(SampleVector test : testList) {
              da.reconstruct(test.getX(), reconstruct_x);
              totalError += ClassVerify.squaredError(test.getX(), reconstruct_x);
            }
            logger.info("Mean square error is " + totalError / testList.size());
    } catch(Throwable e) {
      logger.error("", e);
View Full Code Here

TOP

Related Classes of org.gd.spark.opendl.downpourSGD.Backpropagation.AutoEncoder

Copyright © 2018 www.massapicom. 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.