java - データのラベルを表示する [deep4j]

Question

分類に使用する traindata / testdata のラベルを印刷したいと思います。両方の入力の定義を次に示します (deep4j を使用)。

    InputSplit[] inputSplit = fileSplit.sample(pathFilter, splitTrainTest, 1 - splitTrainTest);
    InputSplit trainData = inputSplit[0];
    InputSplit testData = inputSplit[1];

次に、次のように DataSetIterator に変換されます。

    ImageRecordReader recordReader = new ImageRecordReader(height, width, channels, labelMaker);
    recordReader.initialize(trainData, null);
    trainIter = new RecordReaderDataSetIterator(recordReader, batchSize, 1, numLabels);

次に、この関数の各反復子で見つかったラベルごとの例の数を出力したいと思います。

public void print(DataSetIterator iter){

    HashMap<String, Integer> hash = new HashMap<String, Integer>();

    while(iter.hasNext()){
        DataSet example = iter.next();
        for(int i = 0 ; i<numLabels ; i++){
            if(example.getLabels().getDouble(i)==1.){
                String label = example.getLabelName(i);
                if(hash.containsKey(label))
                    hash.put(label, hash.get(label)+1);
                else
                    hash.put(label, 1);
            }
        }
    }

    for (String label: hash.keySet()){
        System.out.println("   label : " + label.toString() + ", " + hash.get(label) + " examples");
    }
}

問題は、ラベルごとに 1 つの例しか表示されないのに対し、もっと多くの例が表示されることです...そしてfileSplit.sample()、関数を使用してデータセットを分割しないと、適切な数の例が表示されます。なにか提案を ？