如何一起使用k折的交叉验证和加权随机采样器？

Question

我正在尝试使用pytorch（data_loader）训练我的分类模型（3类） 因此，我的第一个问题是数据没有平衡，所以我使用了加权随机采样器：

  class_weights = [5,1,1] 

  sample_weights = [0] * len(ds)

  for idx in enumerate(ds):
      class_weight = class_weights[idx[1]['targets']]
      sample_weights[idx[0]] = class_weight

  sampler = WeightedRandomSampler(sample_weights, num_samples=
                                    len(sample_weights), replacement=True)
  DataLoader( ds, batch_size=batch_size,  sampler=sampler, num_workers=2 )

但是现在我还有另一个问题，即过度拟合，所以我想使用 k折的交叉验证

有人可以帮助我，我真的不知道该怎么做！