Sklearn 단순회귀분석

1 개요[ | ]

sklearn 단순선형회귀분석

2 예시 1: 키와 몸무게[ | ]

Python

CPU

1.5s

MEM

101M

1.5s

Copy

import pandas as pd
df = pd.DataFrame({
'height': [1.47, 1.50, 1.52, 1.55, 1.57, 1.60, 1.63, 1.65, 1.68, 1.70, 1.73, 1.75, 1.78, 1.80, 1.83],
'mass': [52.21, 53.12, 54.48, 55.84, 57.20, 58.57, 59.93, 61.29, 63.11, 64.47, 66.28, 68.10, 69.92, 72.19, 74.46],
})

X = df[['height']]
Y = df['mass']

from sklearn import linear_model
reg = linear_model.LinearRegression()
reg.fit(X, Y)
print( "coefficient=", reg.coef_ )
print( "intercept=", reg.intercept_ )
print( "R²=", reg.score(X, Y) )

coefficient= [61.27218654]
intercept= -39.06195591884392
R²= 0.9891969224457968

→ 회귀식 [math]\displaystyle{ y=61.27218654x-39.06195591884392 }[/math]

→ 결정계수 [math]\displaystyle{ R^2=0.9891969224457968 }[/math]

3 예시 2: 아이스티 주문[ | ]

Python

CPU

1.5s

MEM

103M

1.6s

Copy

import pandas as pd
df = pd.read_csv('https://raw.githubusercontent.com/jmnote/z-dataset/master/simple-regression/iced-tea-orders.csv')
print( df )

X = df[['high_temperature']]
Y = df['order']

from sklearn.linear_model import LinearRegression
reg = LinearRegression().fit(X, Y)
print( "coefficient=", reg.coef_ )
print( "intercept=", reg.intercept_ )
print( "R²=", reg.score(X, Y) )

          date weekday  high_temperature  order
0   2002-07-22     Mon                29     77
1   2002-07-23     Tue                28     62
2   2002-07-24     Wed                34     93
3   2002-07-25     Thr                31     84
4   2002-07-26     Fri                25     59
5   2002-07-27     Sat                29     64
6   2002-07-28     Sun                32     80
7   2002-07-29     Mon                31     75
8   2002-07-30     Tue                24     58
9   2002-07-31     Wed                33     91
10  2002-08-01     Thr                25     51
11  2002-08-02     Fri                31     73
12  2002-08-03     Sat                26     65
13  2002-08-04     Sun                30     84
coefficient= [3.73788546]
intercept= -36.361233480176196
R²= 0.8225092881166944

→ 회귀식 [math]\displaystyle{ y=3.73788546x-36.361233480176196 }[/math]

→ 결정계수 [math]\displaystyle{ R^2=0.8225092881166944 }[/math]

4 같이 보기[ | ]