public testOnDataSet ( |
||
nnds | ||
return | int[] |
public int[] testOnDataSet(NNDataSet nnds)
{
int[] result = new int[] { 0, 0 };
nnds.refreshDataset();
while (nnds.hasMoreExamples())
{
NNExample nne = nnds.getExampleAtRandom();
Vector prediction = predict(nne);
if (nne.isCorrect(prediction))
{
result[0] = result[0] + 1;
}
else
{
result[1] = result[1] + 1;
}
}
return result;
}
}
public void testPerceptron() { DataSet irisDataSet = DataSetFactory.getIrisDataSet(); Numerizer numerizer = new IrisDataSetNumerizer(); NNDataSet innds = new IrisNNDataSet(); innds.createExamplesFromDataSet(irisDataSet, numerizer); Perceptron perc = new Perceptron(3, 4); perc.trainOn(innds, 10); innds.refreshDataset(); perc.testOnDataSet(innds); }