다음은 배열로 변환된 열 개의 행을 출력한 결과입니다.
array([[3, 3, 0, 0, 2, 1], [3, 3, 0, 0, 2, 2], [3, 3, 0, 0, 2, 0], [3, 3, 0, 0, 1, 1], [3, 3, 0, 0, 1, 2], [3, 3, 0, 0, 1, 0], [3, 3, 0, 0, 0, 1], [3, 3, 0, 0, 0, 2], [3, 3, 0, 0, 0, 0], [3, 3, 0, 1, 2, 1]], dtype=int8)
이제 torch 모듈을 이용하여 배열을 텐서로 변환합니다.
코드 2-5 배열을 텐서로 변환
categorical_data = torch.tensor(categorical_data, dtype=torch.int64)
categorical_data[:10]
텐서로 변환된 결과에 대한 열 개의 결과를 살펴보면 다음과 같습니다.
tensor([[3, 3, 0, 0, 2, 1], [3, 3, 0, 0, 2, 2], [3, 3, 0, 0, 2, 0], [3, 3, 0, 0, 1, 1], [3, 3, 0, 0, 1, 2], [3, 3, 0, 0, 1, 0], [3, 3, 0, 0, 0, 1], [3, 3, 0, 0, 0, 2], [3, 3, 0, 0, 0, 0], [3, 3, 0, 1, 2, 1]])