format

src/edu/hitsz/c102c/cnn/RunCNN.java

@@ -10,6 +10,7 @@
 public class RunCNN {
 
 	public static void runCnn() {
+		//创建一个卷积神经网络
 		LayerBuilder builder = new LayerBuilder();
 		builder.addLayer(Layer.buildInputLayer(new Size(28, 28)));
 		builder.addLayer(Layer.buildConvLayer(6, new Size(5, 5)));
@@ -18,18 +19,18 @@ public static void runCnn() {
 		builder.addLayer(Layer.buildSampLayer(new Size(2, 2)));
 		builder.addLayer(Layer.buildOutputLayer(10));
 		CNN cnn = new CNN(builder, 50);
+
+		//导入数据集
 		String fileName = "dataset/train.format";
 		Dataset dataset = Dataset.load(fileName, ",", 784);
-		cnn.train(dataset, 100);//
+		cnn.train(dataset, 3);//
 		String modelName = "model/model.cnn";
-		cnn.saveModel(modelName);
-		// CNN cnn = CNN.loadModel(modelName);
-		// Dataset dataset = Dataset.load(fileName,
-		// ",", 784);
-		// cnn.train(dataset, 100);
-		// cnn.saveModel(modelName);
+		cnn.saveModel(modelName);		
 		dataset.clear();
 		dataset = null;
+
+		//预测
+		// CNN cnn = CNN.loadModel(modelName);	
 		Dataset testset = Dataset.load("dataset/test.format", ",", -1);
 		cnn.predict(testset, "dataset/test.predict");
 	}