MachineLearning
특정 상관계수 이하 제거 및 x,y 분리
forkballpitch
2022. 8. 31. 23:50
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']