小知识点【pytorch dataloader划分数据集】

最新推荐文章于 2024-03-31 11:33:08 发布

� rk

最新推荐文章于 2024-03-31 11:33:08 发布

阅读量279

点赞数 1

分类专栏： pytorch+python 小知识点文章标签： pytorch 深度学习机器学习

本文链接：https://blog.csdn.net/renkke/article/details/125321615

版权

pytorch+python 小知识点专栏收录该内容

8 篇文章 0 订阅

订阅专栏

项目场景：

描述：pytorch生成图片，已经将数据导入到dataloader，然后怎么划分数据集

下面展示一些 代码片。

    import torch
    import torchvision
    import torchvision.transforms as transforms
    
    path='D:/horse'
    dataset=torchvision.datasets.ImageFolder(path,transform=transform)
    train_size = int(0.8 * len(dataset))
    test_size = len(dataset) - train_size
    train_dataset, test_dataset = torch.utils.data.random_split(dataset, [train_size, test_size])