加载训练数据时Dataset的__getitem__方法报错raise KeyError(key) from err KeyError: 193

将模型改为交叉验证后,数据加载时提示KeyError:193,

猜测是调用__getitem__时没有找到索引为193的样本,几次排查后发现是使用KFold划分交叉验证的训练集和验证集后,没有重置索引,这里的193报错应该是索引为193的样本不在训练集的DataLoader中,而是在测试集中

解决方法,在划分训练集和验证集的代码后加两行重置索引的代码即可

train_df_fold = train_df.iloc[train_ids]
test_df_fold = train_df.iloc[test_ids]
# 重置索引
train_df_fold = train_df_fold.reset_index(drop=True)
test_df_fold = test_df_fold.reset_index(drop=True)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
__setitem__和__getitem__是Python中的魔术方法(magic method),用于支持对象的索引访问和赋值操作。 __getitem__方法用于实现获取对象的索引值,即通过下标获取对象中的元素。当我们使用索引来访问一个对象的元素,实际上是调用了该对象的__getitem__方法,并将索引值作为参数传递给该方法。这个方法需要返回相应索引的值。 例如,我们可以定义一个类MyList,其中定义了一个列表属性data,并重写了__getitem__方法,使其支持通过下标访问列表元素: ``` class MyList: def __init__(self): self.data = [1, 2, 3] def __getitem__(self, index): return self.data[index] mylist = MyList() print(mylist[0]) # 输出1 ``` __setitem__方法用于实现给对象赋值,即通过下标给对象的某个元素赋值。当我们使用索引来对一个对象的元素进行赋值,实际上是调用了该对象的__setitem__方法,并将索引值和赋值的值作为参数传递给该方法。这个方法不需要返回任何值。 例如,我们可以继续在上述的MyList类中添加__setitem__方法,使其支持通过下标给列表元素赋值: ``` class MyList: def __init__(self): self.data = [1, 2, 3] def __getitem__(self, index): return self.data[index] def __setitem__(self, index, value): self.data[index] = value mylist = MyList() mylist[0] = 4 print(mylist[0]) # 输出4 ``` 通过重写__getitem__和__setitem__方法,我们可以实现自定义的索引访问和赋值逻辑,使得我们的对象可以像列表或字典一样进行索引操作。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值