C++ (Cpp) MNISTDataset::getFeatureNumber Examples

Example #1

File: LogisticModel.cpp Project: roles/deep_learning

void testMNISTTraining(){
    MNISTDataset mnist;
    mnist.loadData();

    Logistic logi(mnist.getFeatureNumber(), mnist.getLabelNumber());
    logi.setModelFile("result/LogisticModel.dat");
    TrainModel logisticModel(logi);
    logisticModel.train(&mnist, 0.01, 10, 1000);
}

Example #2

File: LogisticModel.cpp Project: roles/deep_learning

void testMNISTGuassianTraining(){
    MNISTDataset mnist;
    mnist.loadData();
    mnist.rowNormalize();

    Logistic logi(mnist.getFeatureNumber(), mnist.getLabelNumber());
    logi.setModelFile("result/LogisticModel.dat");
    TrainModel logisticModel(logi);
    logisticModel.train(&mnist, 0.13, 500, 100);
}

Example #3

File: MLPModel.cpp Project: witgo/deep_learning

void testMNIST(){
    MNISTDataset mnist;
    mnist.loadData();
    MLP mlp; 

    SigmoidLayer *firstLayer = new SigmoidLayer(mnist.getFeatureNumber(), 500);
    Logistic *secondLayer = new Logistic(500, mnist.getLabelNumber());
    mlp.addLayer(firstLayer);
    mlp.addLayer(secondLayer);
    mlp.setModelFile("result/MLPModel.dat");

    TrainModel mlpModel(mlp);
    mlpModel.train(&mnist, 0.01, 20, 1);
}