분류 전체보기194 파일생성 submission = pd.read_csv('submission.csv') submission['count']=linear_prediction submission.to_csv('linear.csv', index = False) 2022. 8. 31. 단순 선형회귀 분석 from sklearn.model_selection import KFold from sklearn.linear_model import LinearRegression linear_model_result = [] kf = KFold(n_splits = 5) for idx, (trn_idx, val_idx) in enumerate(kf.split(train_x)): train_x = np.array(train_x) trn_x = train_x[trn_idx] val_x = train_x[val_idx] trn_y = train_y[trn_idx] val_y = train_y[val_idx] linear_model = LinearRegression() linear_model.fit(trn_x, trn_y) li.. 2022. 8. 31. 특정 상관계수 이하 제거 및 x,y 분리 drop_list=[] corr = train.corr()['count'] drop_list.extend(corr[abs(corr) 2022. 8. 31. 상관계수 heatmap plt.figure(figsize = 10,10)) sns.heatmap(train.corr()) 2022. 8. 31. 이전 1 2 3 4 5 ··· 49 다음