사과 선별장의 선별기 측정 데이터이다. 무게·당도·단단함 등 측정값으로 그 사과가 상품 등급인지(grade_ok=1) 아닌지(grade_ok=0)를 예측하는 이진 분류 문제이다. 학습용 데이터 apple_train.csv로 모델을 학습하고, 평가용 데이터 apple_test.csv의 각 사과에 대한 상품 등급 여부를 예측하여 result.csv 파일을 생성하는 코드를 작성하여 제출하시오.
[제출 형식]
result.csv (별도 디렉토리 지정 금지)pred (예측한 상품 등급 여부 0 또는 1)pred 1개 (자동 생성되는 index 컬럼 제거)[변수 설명]
| 컬럼명 | 설명 | 값 |
|---|---|---|
| weight_g | 무게(g) | 실수 |
| diameter_mm | 지름(mm) | 실수 |
| sugar_brix | 당도(Brix) | 실수 |
| firmness | 과육 단단함(kgf) | 실수 |
| color_score | 착색 점수 | 1 ~ 10 |
| days_stored | 수확 후 보관 일수 | 정수 |
| grade_ok | 상품 등급 여부 (train에만 존재, 예측 대상) | 1: 상품, 0: 비상품 |
💡 이렇게 해 보세요
- 정답 컬럼
grade_ok를train.pop()으로 꺼내 문제(X)와 정답(y)을 나눕니다.- 0/1을 맞히는 분류이므로
RandomForestClassifier로fit→predict합니다.- 예측값을
pd.DataFrame({'pred': 예측값})으로 감싸to_csv('result.csv', index=False)로 저장합니다.- 모델 이름이 기억나지 않으면
import sklearn.ensemble후print(dir(sklearn.ensemble)), 사용법은help(RandomForestClassifier).
■ CSV 파일명: result.csv · 예측 칼럼명: pred (칼럼 1개)
■ 제출 CSV 파일 형식 예시
pred 1 0 1 ……
■ CSV 파일 확인 방법: 생성된 파일을 아래처럼 출력해 확인
import pandas as pd
result = pd.read_csv("result.csv")
print(result.head())파이썬 실행환경 불러오는 중 · 보통 3초