반응형
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']반응형
'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 |