Dataset
Dataset
package mydl.dataset;
import java.util.ArrayList;
import mydl.tensor.Tensor;
public class Test {
public static void main(String[] args) {
ArrayList<Tensor> train_image = MNIST.readTrainImage2D();
System.out.println(train_image.size());
ArrayList<Tensor> train_label = MNIST.readTrainLabel();
System.out.println(train_label.size());
ArrayList<Tensor> test_image = MNIST.readTestImage2D();
System.out.println(test_image.size());
ArrayList<Tensor> test_label = MNIST.readTestLabel();
System.out.println(test_label.size());
}
}
Output:
60000
60000
10000
10000