torch dataset, dataloader 에러가 나네요

dataset code 입니다.

class CustomDataset(Dataset):

def __init__(self, df):
    self.texts = df[['full_text']]
    
def __len__(self):
    len(self.texts)
    
def __getitem__(self, idx):
    test_text = tokenizer(self.texts.loc[idx].values[0], padding='max_length', max_length=512, truncation=True, return_tensors='pt')
    return test_text

돌려지나 확인하려고 해보았는데 ‘NoneType’ object cannot be interpreted as an integer이 뜨네요
nonetype이 없는데.,
class로 안하고 하나하나 받아봤을땐 작동하더라고요
class를 받고 len을 확인해봐도 none이 뜨네요.

좋아요 1

해결했습니다 __len__함수에서
return len(self.texts)를 해야하네요 허허

좋아요 2

해결 축하드립니다~~ :tada:
AS 센터에 가면 갑자기 동작이 잘 되는 전자기기처럼, 질문을 올리고 나면 갑자기 문제가 보이는 경우가 많더라구요! :sweat_smile:

좋아요 2

이 글은 마지막 댓글이 달린지 오래(7일)되어 자동으로 닫혔습니다. 댓글 대신 새로운 글을 작성해주세요! :slight_smile: