在 Python 中,有多种工具可以用来建立回归模型。其中,常见的工具有 scikit-learn、TensorFlow、Keras 等。
下面是使用 scikit-learn 建立回归模型的简单示例:
import numpy as np
from sklearn.linear_model import LinearRegression
# 设置训练数据
x = np.array([[1], [2], [3], [4], [5]])
y = np.array([1, 2, 3, 4, 5])
# 建立回归模型
reg = LinearRegression().fit(x, y