다항회귀

다항 회귀

kaggle_fish_market을 사용하여 다항 회귀를 해 보려 한다.

데이터 전처리

데이터 분리

먼저 훈련과 타겟 데이터를 나누어 준다.

y=df['Species']
X=df.drop('Species', axis=1)

drop을 사용해서 1행에 있는 ‘Species’를 지워 주어서 간단하게 훈련 데이터를 만들 수 있다.

Encoding

그다음 범주형 데이터를 매핑해 주려 한다.

‘Species’는 총 7가지의 물고기 종류인데

sklearn.preprocessing의 LabelEncoder를 사용하려 한다.

encoder = LabelEncoder()
y = encoder.fit_transform(y)
y_mapping = {index: label for index, label in enumerate(encoder.classes_)}

encoder의 fit과 transform을 같이 사용하고 enumerate() 함수를 사용해서 루프를 돌려 주어서 classes_의 0,1,2..값들을 label 값과 매핑시켜 주고

사전형으로 바꾸어서 key에 ‘Species’의 각 index 값을 넣어 주고 value에는 encoder로 매핑해 준 값들을 넣어 주어 사전을 만들어 주었다.