2948 explained code solutions for 75 technologies

python-scikit-learnLinear regression with multiple variables

In order to use multiple feature variables with linear regression we can just use 2-dimensional arrays as in below X variable example:

from sklearn import linear_model

X = [[1,2], [2,1], [2,3], [3,2], [4,5]]
y = [3, 3, 5, 6, 9]

model = linear_model.LinearRegression()
model.fit(X, y)

r2 = model.score(X,y)ctrl + c
from sklearn import

import module from scikit-learn


initialize linear regression model

X =

declare feature dataset with 2 variables and 5 objects


first object variables (we have 2, but it can have any number of variables)


train model with a given features and target variable dataset


trained model R2 score for a given (test) dataset