testdatapersist.java

来自「VHDL制作的ann的code」· Java 代码 · 共 33 行

package org.encog.neural.persist;

import java.io.File;

import junit.framework.TestCase;

import org.encog.neural.data.basic.BasicNeuralDataSet;
import org.encog.neural.data.csv.TestCSVNeuralData;
import org.encog.neural.networks.XOR;

public class TestDataPersist extends TestCase {
	
	public static final String FILENAME = "encogtest.xml";
	
	public void testCSVData() throws Exception 
	{
		BasicNeuralDataSet trainingData = new BasicNeuralDataSet(XOR.XOR_INPUT,XOR.XOR_IDEAL);

		EncogPersistedCollection encog = new EncogPersistedCollection();
		encog.add(trainingData);
		encog.save(TestDataPersist.FILENAME);

		EncogPersistedCollection encog2 = new EncogPersistedCollection();
		encog2.load(TestDataPersist.FILENAME);
		
		BasicNeuralDataSet set = (BasicNeuralDataSet) encog2.getList().get(0);
		
		XOR.testXORDataSet(set);
		
		new File(TestCSVNeuralData.FILENAME).delete();
	}	
}