model.fit() / model.predict()
機械学習モデルにデータを学習させ（fit）、新しいデータに対して予測を行います（predict）。scikit-learnの多くのモデルに共通する書き方です。
構文
model.fit(学習データ, 正解ラベル)
model.predict(新しいデータ)
使用例
from sklearn.linear_model import LinearRegression
model = LinearRegression()
X = [[1], [2], [3]]
y = [10, 20, 30]
model.fit(X, y)
print(model.predict([[4]]))実行結果
[40.]💡 実践的な使用例
from sklearn.linear_model import LinearRegression
model = LinearRegression()
X = [[20], [30], [40]]
y = [5, 7, 9]
model.fit(X, y)
new_rooms = [[25], [35], [45]]
print(model.predict(new_rooms))predict()には1件だけでなく、複数件のデータをまとめてリストで渡すこともでき、一度に複数の予測結果を受け取れます。
関連トピック
AI scikit-learn 機械学習