Advertisement
Guest User

Untitled

a guest
Jan 23rd, 2017
87
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.80 KB | None | 0 0
  1. public UIServer uiServer = null;
  2. public StatsStorage statsStorage = new InMemoryStatsStorage();
  3. public void TrainNeuronNetworkMulti(MultiLayerNetwork model, DataSetIterator[] train_dataset) throws Exception {
  4. ScoreIterationListener train_listener = new ScoreIterationListener(score_by_batch);
  5. model.setListeners(train_listener);
  6. if (use_webservice) {
  7. uiServer = UIServer.getInstance();
  8. model.setListeners(new StatsListener(statsStorage, listenerFrequency));
  9. uiServer.attach(statsStorage);
  10. for (int i = 0; i < numEpochs; i++) {
  11. for (int r = 0; r < train_dataset.length; r++) {
  12. model.fit(train_dataset[r]);
  13. }
  14. uiServer.stop();
  15. }
  16. } else {
  17.  
  18. for (int i = 0; i < numEpochs; i++) {
  19. for (int r = 0; r < train_dataset.length; r++) {
  20. model.fit(train_dataset[r]);
  21. }
  22. }
  23. }
  24.  
  25. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement