[Pytorch] dataloader key error

cs2tree·2023년 3월 16일
0

Pytorch

목록 보기
3/3

판다스를 데이터 로더에 넣었을 때 생기는 문제이다.

df = pd.DataFrame({"a:[1,2,3], "b":[4,5,6]}); 

이렇게 데이터 프레임이 정의된 경우 df[0]을 인덱싱할 수 없기 때문에 발생하는 오류이다.

이를 해결하기 위해서는 넘파이 어레이로 바꿔주면 된다.

df.to_numpy()

profile
___호기심 많은 씨앗___ 씨앗이 나무가 될 때까지 :D

0개의 댓글