Merge pull request deeplearning4j#1301 from deeplearning4j/ab_evalts

Fix MultiLayerNetwork.evaluate(DataSetIterator) for time series
jungleTail · Mar 19, 2016 · 266c498 · 266c498
2 parents 2e18d83 + 6333cd1
commit 266c498
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/multilayer/MultiLayerNetwork.java b/deeplearning4j-core/src/main/java/org/deeplearning4j/nn/multilayer/MultiLayerNetwork.java
@@ -2264,7 +2264,8 @@ public Evaluation evaluate(DataSetIterator iterator, List<String> labelsList){
                 }
             } else {
                 out = this.output(features,false);
-                e.eval(labels,out);
+                if(labels.rank() == 3 ) e.evalTimeSeries(labels,out);
+                else e.eval(labels,out);
             }
         }