Python for AI 실습게시판

				1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
			#
# 따라하며 배우는 파이썬과 데이터과학(생능출판사 2020)
# 14.12 당뇨병 예제를 학습 데이터와 테스트 데이터로 구분하자, 381쪽
#
import matplotlib.pyplot as plt
import numpy as np
from sklearn.linear_model import LinearRegression
from sklearn import datasets
from sklearn.model_selection import train_test_split
 
# 당뇨병 데이터 세트를 sklearn의 데이터집합으로부터 읽어들인다.
diabetes = datasets.load_diabetes()
 
X_train, X_test, y_train, y_test = train_test_split(diabetes.data, diabetes.target,
                                                    test_size = 0.2)
regr = LinearRegression()
 
regr.fit(X_train, y_train)
y_pred = regr.predict(X_test) # 테스트 데이터로 예측해보자. 
 
mse_set = (y_pred - y_test) ** 2
sum_mse = 0.0
for value in mse_set:
    sum_mse += value
 
mse = sum_mse / len(mse_set)
 
print('평균제곱 오차(MSE):', mse)
Colored by Color Scripter
			cs		