@Test public void when_getBatchSizeIsCalled_expect_batchSizeIsReturned() { INDArray features = Nd4j.create(5, 10); FeaturesLabels sut = new FeaturesLabels(features); long batchSize = sut.getBatchSize(); assertEquals(5, batchSize); }
public long getBatchSize() { return features.shape()[0]; }
FeaturesLabels { public long getBatchSize() { return features.shape()[0]; } }
FeaturesLabels { public long getBatchSize() { return features.shape()[0]; } FeaturesLabels(INDArray features); }
FeaturesLabels { public long getBatchSize() { return features.shape()[0]; } FeaturesLabels(INDArray features); long getBatchSize(); void putLabels(String name, INDArray labels); INDArray getLabels(String name); }
FeaturesLabels { public long getBatchSize() { return features.shape()[0]; } FeaturesLabels(INDArray features); long getBatchSize(); void putLabels(String name, INDArray labels); INDArray getLabels(String name);  }