Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- public UIServer uiServer = null;
- public StatsStorage statsStorage = new InMemoryStatsStorage();
- public void TrainNeuronNetworkMulti(MultiLayerNetwork model, DataSetIterator[] train_dataset) throws Exception {
- ScoreIterationListener train_listener = new ScoreIterationListener(score_by_batch);
- model.setListeners(train_listener);
- if (use_webservice) {
- uiServer = UIServer.getInstance();
- model.setListeners(new StatsListener(statsStorage, listenerFrequency));
- uiServer.attach(statsStorage);
- for (int i = 0; i < numEpochs; i++) {
- for (int r = 0; r < train_dataset.length; r++) {
- model.fit(train_dataset[r]);
- }
- uiServer.stop();
- }
- } else {
- for (int i = 0; i < numEpochs; i++) {
- for (int r = 0; r < train_dataset.length; r++) {
- model.fit(train_dataset[r]);
- }
- }
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement