DJL-Java开发者动手学深度学习之图片分类

最新推荐文章于 2024-06-13 07:42:00 发布

ousinka

最新推荐文章于 2024-06-13 07:42:00 发布

阅读量1.4k

点赞数 1

分类专栏： Java开发者动手学习深度学习 d2lcoder DJL 文章标签： java 深度学习分类

本文链接：https://blog.csdn.net/xxcnb/article/details/122386126

版权

之前，我们文章里有讲，通过softmax回归对图片进行分类，具体文章请见《使用Softmax进行分类代码实现》。今天我们通过高级API更简洁地实现多层感知机。

准备数据集

private static RandomAccessDataset getDataset(Dataset.Usage usage)
    throws IOException {
   
    Mnist mnist =
        Mnist.builder()
        .optUsage(usage)
        .setSampling(32, true)
        .optLimit(64)
        .build();
    mnist.prepare(new ProgressBar());
    return mnist;
}

//训练集
RandomAccessDataset trainingSet = getDataset(Dataset.Usage