python-scikit-learnGet linear regression model coefficients
import numpy as np
from sklearn import datasets, linear_model
X, y = datasets.load_diabetes(return_X_y=True)
model = linear_model.LinearRegression()
model.fit(X, y)
coefs = model.coef_ctrl + cfrom sklearn importimport module from scikit-learn | import numpyimport Numpy module |
datasets.load_diabetesloads sample diabetes database | linear_model.LinearRegressioninitialize linear regression model |
.fit(train model with a given features and target variable dataset | coefslist of trained coefficients |
Usage example
import numpy as np
from sklearn import datasets, linear_model
X, y = datasets.load_diabetes(return_X_y=True)
model = linear_model.LinearRegression()
model.fit(X, y)
coefs = model.coef_
print(coefs)
output
Intercept: 152.13348416289597
Coefs: [ -10.0098663 -239.81564367 519.84592005 324.3846455 -792.17563855
476.73902101 101.04326794 177.06323767 751.27369956 67.62669218]
R2: 0.5177484222203498
Related
More of Python Scikit Learn
- Using Ridge regression model example
- Use imputer for one column only
- How to inverse transform PCA reduced data
- How to disable cross validation in grid search CV
- Neural network example
- Grid search CV example
- Binary classification example
- PCA dimensionality reduction example
- Decision tree classifier example
- Jaccard score
See more codes...