溫馨提示×

Scikit-learn中怎么實現線性回歸

小億
159
2024-05-10 17:07:52
欄目: 編程語言

在Scikit-learn中,可以使用LinearRegression類來實現線性回歸。下面是一個簡單的示例代碼:

from sklearn.linear_model import LinearRegression
import numpy as np

# 創建一些示例數據
X = np.array([[1], [2], [3], [4]])
y = np.array([2, 4, 6, 8])

# 創建線性回歸模型
model = LinearRegression()

# 擬合模型
model.fit(X, y)

# 進行預測
y_pred = model.predict([[5]])

print(y_pred)

在這個示例中,首先創建了一些示例數據X和y,然后創建了一個LinearRegression模型,并用fit方法擬合了這些數據。最后使用predict方法對新的數據進行預測。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女