In [21]:
# from mlwpy import *로 가려진 # 라이브러리 import 구문은 다음과 같습니다 # from sklearn import (datasets, neighbors, # model_selection as skms, # linear_model, metrics) # 자세한 내용은 부록 A를 참고하세요 linreg = linear_model.LinearRegression() diabetes = datasets.load_diabetes() scores = [] for r in range(10): tts = skms.train_test_split(diabetes.data, diabetes.target, test_size=.25) (diabetes_train_ftrs, diabetes_test_ftrs, diabetes_train_tgt, diabetes_test_tgt) = tts fit = linreg.fit(diabetes_train_ftrs, diabetes_train_tgt) preds = fit.predict(diabetes_test_ftrs) score = metrics.mean_squared_error(diabetes_test_tgt, preds) scores.append(score) scores = pd.Series(np.sqrt(sorted(scores))) df = pd.DataFrame({'RMSE':scores}) df.index.name = 'Repeat' display(df.T)
Repeat |
0 |
1 |
2 |
3 |
4 |
5 |
6 |
7 |
8 |
9 |
RMSE |
49.00 |
50.19 |
51.97 |
52.07 |
53.20 |
55.70 |
56.25 |
57.49 |
58.64 |
58.69 |