Python中的多元线性回归 - 在MATLAB中使用mvregress的模拟?

时间:2016-05-02 14:32:31

标签: python matlab numpy scikit-learn linear-regression

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

1 个答案:

答案 0 :(得分:2)

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.]])