Pytorch:torch.utils.data.random_split()

random_split() 函数说明:

torch.utils.data.random_split(dataset, lengths, generator=<torch._C.Generator object>)

参数:

  • dataset(Dataset) -要拆分的数据集
  • lengths(序列) -要产生的分割长度
  • generator(torch.Generator) -用于随机排列的生成器。

注:关于torch.Generator详见笔记:Pytorch:torch.Generator()

pytorch: random_split(),函数的具体定义如下:

def random_split(dataset, lengths):
    r"""
    Randomly split a dataset into non-overlapping new datasets of given lengths.

    Arguments:
        dataset (Dataset): Dataset to be split
        lengths (sequence): lengths of splits to be produced
    """
    if sum(lengths) != len(dataset):
        raise ValueError("Sum of input lengths does not equal the length of the input dataset!")

    indices = randperm(sum(lengths)).tolist()
    return [Subset(dataset, indices[offset - length:offset]) for offset, length in zip(_accumulate(lengths), lengths)]

以U-Net代码(详见:U-Net代码复现)为例:

n_val = int(len(dataset) * val_percent)
n_train = len(dataset) - n_val
train_set, val_set = random_split(dataset, [n_train, n_val], generator=torch.Generator().manual_seed(0))

通过random_split()将数据分为训练集和验证集(随机)

相关推荐

最近更新

  1. WebKit源代码探秘:深入理解其组织结构与组件

    2023-12-06 14:32:03       0 阅读
  2. 【回溯+双指针算法题记录】回文字符串汇总

    2023-12-06 14:32:03       0 阅读
  3. 2288. 价格减免

    2023-12-06 14:32:03       1 阅读
  4. Quartz 介绍

    2023-12-06 14:32:03       1 阅读
  5. Taro自定义实现本地路径转换为文件

    2023-12-06 14:32:03       1 阅读
  6. Python 类与对象:深入理解与应用

    2023-12-06 14:32:03       0 阅读

热门阅读

  1. VB.NET二维数组的组合

    2023-12-06 14:32:03       48 阅读
  2. 通俗讲解分布式锁:场景和使用方法

    2023-12-06 14:32:03       31 阅读
  3. 2312skia,12画布包与路径包

    2023-12-06 14:32:03       31 阅读
  4. 大数据(十一):概率统计基础

    2023-12-06 14:32:03       41 阅读
  5. 关于SQL注入问题及解决--小记

    2023-12-06 14:32:03       33 阅读
  6. html复习

    2023-12-06 14:32:03       34 阅读
  7. Python中的global关键字

    2023-12-06 14:32:03       35 阅读
  8. MIT 6.S081学习笔记(第五章)

    2023-12-06 14:32:03       28 阅读
  9. js实现tab栏切换

    2023-12-06 14:32:03       35 阅读
  10. npm、yarn、pnpm和cnpm优缺点

    2023-12-06 14:32:03       32 阅读