Skip to content

Instantly share code, notes, and snippets.

lilalinux

Block or report user

Report or block lilalinux

Hide content and notifications from this user.

Learn more about blocking users

Contact Support about this user’s behavior.

Learn more about reporting abuse

Report abuse
View GitHub Profile
View VisualizationEarlyStoppingListener.java
net.rnnClearPreviousState();
trainDataIter.reset();
INDArray pred = null;
List<INDArray> predictionsTrain = new ArrayList<>();
while (trainDataIter.hasNext()) {
DataSet t = trainDataIter.next();
INDArray featureMatrix = t.getFeatureMatrix(); // [miniBatchSize,inputSize,inputTimeSeriesLength], 5000x2x20 -> [3,5000,2,20,1,100000,5000,0,-1,99]
INDArray predictionMiniBatch = net.rnnTimeStep(featureMatrix);
normalizer.revertLabels(predictionMiniBatch);
You can’t perform that action at this time.