Как использовать Java Encog с обучающим методом GA для решения XOR?
Мой вывод очень плохой. На самом деле не равно идеал. Какие коды неверны?
Мой вывод:
Epoch #129 Error:8.755514431853456E-6
Neural Network Results:
0.0,0.0, actual=0.57600,ideal=0.0
1.0,0.0, actual=0.58016,ideal=1.0
0.0,1.0, actual=0.58886,ideal=1.0
1.0,1.0, actual=0.59317,ideal=0.0
Вот мой код:
public class XOR {
public static double XOR_INPUT[][] = { { 0.0, 0.0 }, { 1.0, 0.0 }, { 0.0, 1.0 }, { 1.0, 1.0 } };
/**
* The ideal data necessary for XOR.
*/
public static double XOR_IDEAL[][] = { { 0.0 }, { 1.0 }, { 1.0 }, { 0.0 } };
//public static BasicNetwork network = new BasicNetwork();
public static BasicNetwork createNetwork()
{
// create a neural network, without using a factory
BasicNetwork network = new BasicNetwork();
network.addLayer(new BasicLayer(null,true,2));
network.addLayer(new BasicLayer(new ActivationSigmoid(),true,3));
network.addLayer(new BasicLayer(new ActivationSigmoid(),false,1));
network.getStructure().finalizeStructure();
network.reset();
return network;
}
/**
* The main method.
* @param args No arguments are used.
*/
public static void main(final String args[]) {
BasicNetwork network = createNetwork();
MLDataSet trainingSet = new BasicMLDataSet(XOR_INPUT, XOR_IDEAL);
// train the neural network
CalculateScore score = new TrainingSetScore(trainingSet);
MLTrain train = new MLMethodGeneticAlgorithm(new MethodFactory(){
@Override
public MLMethod factor() {
final BasicNetwork result = createNetwork();
((MLResettable)result).reset();
return result;
}}, score, 500);
int epoch = 1;
do {
train.iteration();;
System.out
.println("Epoch #" + epoch + " Error:" + train.getError());
epoch++;
} while( train.getError() > 0.01);
// test the neural network
System.out.println("Neural Network Results:");
for(MLDataPair pair: trainingSet ) {
final MLData output = network.compute(pair.getInput());
System.out.println(pair.getInput().getData(0) + "," + pair.getInput().getData(1)
+ ", actual=" + output.getData(0) + ",ideal=" + pair.getIdeal().getData(0));
}
}
}
1 ответ
Тренер генетического алгоритма немного отличается от других тренеров в Encog. Это строит популяцию нейронных сетей. Нейронная сеть, которую вы передали, является просто шаблоном того, сколько у вас скрытых слоев и как выглядят слои ввода / вывода. Этот шаблон сети на самом деле не модифицируется обучением, просто население. Как только вы закончите обучение, вам нужно получить лучшую нейронную сеть в популяции. Есть несколько способов сделать это, но самый простой - просто вызвать train.getMethod(). Добавьте следующую строку в ваш код, и он будет работать:
} while( train.getError() > 0.01);
network = (BasicNetwork)train.getMethod(); // Add this
// test the neural network