1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# 必要モジュールをインポート from sklearn.linear_model import LinearRegression from sklearn.datasets import load_wine from sklearn.model_selection import train_test_split data = load_wine() # データを教師用とテスト用に train_X, test_X, train_y, test_y = train_test_split( data.data, data.target, random_state=42) print(data) # 学習器の構築 model = LinearRegression() # 教師データを用いて学習器に学習 model.fit(train_X, train_y) # テスト用データを用いて学習結果をpred_yに代入 pred_y = model.predict(test_X) # 予測結果を出力 #print(pred_y) |