package?info.hb.ccn.main;

import?info.hb.cnn.core.CNN;
import?info.hb.cnn.core.CNN.layerBuilder;
import?info.hb.cnn.core.layer;
import?info.hb.cnn.core.layer.Size;
import?info.hb.cnn.data.DataSet;
import?info.hb.cnn.utils.ConcurentRunner;

public?class?CNNMnist?{

	private?static?final?String?MODEL_NAME?=?“mnist/model/model.cnn“;

	private?static?final?String?TRAIN_DATA?=?“mnist/train.format“;

	private?static?final?String?TEST_DATA?=?“mnist/test.format“;

	private?static?final?String?TEST_PREDICT?=?“mnist/test.predict“;

	public?static?void?main（String[]?args）?{

		System.err.println（“訓(xùn)練階段：“）;
		runTrain（）;
		System.err.println（“測(cè)試階段：“）;
		runTest（）;
		ConcurentRunner.stop（）;

	}

	public?static?void?runTrain（）?{
		//?構(gòu)建網(wǎng)絡(luò)層次結(jié)構(gòu)
		layerBuilder?builder?=?new?layerBuilder（）;
		builder.addlayer（layer.buildInputlayer（new?Size（28?28）））;?//?輸入層輸出map大小為28×28
		builder.addlayer（layer.buildConvlayer（6?new?Size（5?5）））;?//?卷積層輸出map大小為24×2424=28+1-5
		builder.addlayer（layer.buildSamplayer（new?Size（2?2）））;?//?采樣層輸出map大小為12×1212=24/2
		builder.addlayer（layer.buildConvlayer（12?new?Size（5?5）））;?//?卷積層輸出map大小為8×88=12+1-5
		builder.addlayer（layer.buildSamplayer（new?Size（2?2）））;?//?采樣層輸出map大小為4×44=8/2
		builder.addlayer（layer.buildOutputlayer（10））;
		CNN?cnn?=?new?CNN（builder?10）;
		//?加載訓(xùn)練數(shù)據(jù)
		DataSet?dataset?=?DataSet.load（TRAIN_DATA?““?784）;
		//?開始訓(xùn)練模型
		cnn.train（dataset?5）;
		//?保存訓(xùn)練好的模型
		cnn.saveModel（MODEL_NAME）;
		dataset.clear（）;
	}

	public?static?void?runTest（）?{
		//?加載訓(xùn)練好的模型
		CNN?cnn?=?CNN.loadModel（MODEL_NAME）;
		//?加載測(cè)試數(shù)據(jù)
		DataSet?testSet?=?DataSet.load（TEST_DATA?““?-1）;
		//?預(yù)測(cè)結(jié)果
		cnn.predict（testSet?TEST_PREDICT）;
		testSet.clear（）;
	}

}