04/22/04

This sample code illustrates how to use the trainer.

The example pre-supposes a method to obtain the example set and the network. This example adds two epoch event listeners. One stops training when the error drops below 0.05 and the other stops taining at a maximum of 1000 iterations.

Home

// A default network trainer. Since no explicit error manager
// information is set, it assumes sum of squares error under the default
// name. The Stop on error listener assumes the default name for sum
// of squares.
public static void main(String[] args) {
  ExampleSet examples = createExampleSet();
  Network network = makeNetwork();

  Trainer trainer = new Trainer();
  trainer.addEpochEventListener(new StopOnErrorListener(0.05));
  trainer.addEpochEventListener(new MaxIterationsListener(1000));
  trainer.setTrainingExamples(examples);
  try {
    trainer.train(network);
  } catch (TrainingException e) {
    // TODO Auto-generated catch block
    e.printStackTrace();
  }
  System.out.println("Trained network in " + trainer.getEpochCount() + " epochs");
  System.out.println("With an error of " + trainer.getLastError() * 100 + "%");
}