더북(TheBook)

2.4 파이토치 코드 맛보기

파이토치를 사용하여 간단한 분류 및 회귀 모델을 개발하는 방법을 살펴보겠습니다. 그대로 따라 해 보세요.

‘이 책의 활용법’에서 안내한 대로 예제 파일을 내려받은 후 chap02\data 폴더의 car_evaluation.csv 파일을 데이터셋으로 사용합니다. 데이터셋을 열어 보면 다음과 같이 특성(칼럼) 일곱 개로 구성되어 있습니다.

1. price(자동차 가격)

2. maint(자동차 유지 비용)

3. doors(자동차 문 개수)

4. persons(수용 인원)

5. lug_capacity(수하물 용량)

6. safety(안전성)

7. output(차 상태): 이 데이터는 unacc(허용 불가능한 수준) 및 acc(허용 가능한 수준), 양호(good) 및 매우 좋은(very good, vgood) 중 하나의 값을 갖습니다.

 

이때 1~6의 칼럼 정보를 이용하여 일곱 번째 칼럼(차 상태)을 예측하는 코드를 구현해 보겠습니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.