Python中的多元线性回归-MATLAB中mvregression的模拟

Question

我想在Python中使用与MATLAB中的mvregression相同的函数或方法。例如，我们有x1, x2, x3, x4, x5, x6输入和y1, y2, y3输出。在使用这个函数之后，我们应该得到一些估计回归系数。Python有这种能力吗？

Answer 1

sklearn.linear_model.LinearRegression适用于多元线性回归（其中输出是向量，而不是标量）

>>> from sklearn import linear_model
>>> X = [[1, 2, 3, 4, 5, 6]]
>>> Y = [[1, 2, 3]]
>>> lr = linear_model.LinearRegression()
>>> model = lr.fit(X, Y)
>>> model.predict([[1,2,3,4,5,6]])
array([[ 1.,  2.,  3.]])