본문 바로가기
MachineLearning

특정 상관계수 이하 제거 및 x,y 분리

by forkballpitch 2022. 8. 31.
728x90
728x90
drop_list=[]
corr = train.corr()['count']
drop_list.extend(corr[abs(corr)<0.05].index.tolist())
drop_list

linear_train = train.drop(drop_list, axis = 1)
linear_test = test.drop(drop_list, axis = 1)

train_x = linear_train.drop('count', axis =1)
train_y = linear_train['count']
728x90
728x90

'MachineLearning' 카테고리의 다른 글

파일생성  (0) 2022.08.31
단순 선형회귀 분석  (0) 2022.08.31
상관계수 heatmap  (0) 2022.08.31
histplot  (0) 2022.08.31
arm64에 conda 설치하기  (0) 2019.08.16