如何让 torch.utils.data.Subset 从自定义的 torch.utils.data.Dataset 继承方法？答案

【问题标题】：How to make torch.utils.data.Subset inherit methods from customly defined torch.utils.data.Dataset?如何让 torch.utils.data.Subset 从自定义的 torch.utils.data.Dataset 继承方法？
【发布时间】：2022-08-19 06:26:11
【问题描述】：

我已经使用normalize_features(self) 方法定义了我的自定义torch 数据集，它只是对特征进行了规范化。稍后在代码中，我使用torch.utils.data.random_split 将我的数据拆分为训练和测试。但是，这将创建一个Subset 类类型。

现在，当我调用train_data.normalize_features() 时，它会抛出一个错误，因为Subset 没有继承自我可怜的自定义数据集类。另一方面，如果我调用train_data.dataset.normalize_features()，它将访问原始数据集，现在我将得到整个数据集的标准化，尽管只需要对 train_data 进行标准化。

有什么建议可以让Subset 继承我的方法吗？或者有什么巧妙的技巧可以解决这个问题？

你能找到解决方案吗？

标签： inheritance dataset subset torch

【解决方案1】：

我通过将方法变成一个以子集作为参数的函数来解决这个问题

【讨论】：