単回帰分析した結果を配列に格納したい

実現したいこと

dataframeに対して単回帰分析を行い、傾きと切片をそれぞれnp配列に格納したい

前提

機械学習で傾きと切片を予測したく、そのためのデータを用意したい。

発生している問題・エラーメッセージ

エラーはないが、実行しても1行目までしか計算してくれない。
期待しているのはdf[0:i]のiで区切りをずらしていきながら単回帰分析するというもの

該当のソースコード

python

1from sklearn.linear_model import LinearRegression 2import pandas as pd 3import numpy as np 4 5df = pd.read_csv('/content/drive/MyDrive/Data/study/data-3.csv')6 7last = len(df)8data = []9 10for i in range(1,last):11 df = df.iloc[0:i]12 df.head()13 14 x_d = df[['x']]15 y_d = df[['y']]16 17 print(x_d)18 print(y_d)19 20 model_lr = LinearRegression()21 model_lr.fit(x_d, y_d)22 print(model_lr.coef_)23 print(model_lr.intercept_)24 y = model_lr.coef_ * i + model_lr.intercept_ 25 print("y=",model_lr.coef_ * i + model_lr.intercept_)26 data.append(y)27 28 np_data = np.array(data)29 coef = np.vstack(model_lr.coef_)30 intercept = np.vstack(model_lr.intercept_)

試したこと

a[0]は出力され、i=2の時のa[1]のprintも確認できています。

コメントを投稿

0 コメント