【发布时间】:2017-09-22 08:46:01
【问题描述】:
我正在尝试运行 Elastic Net 回归,但出现以下错误:NameError: name 'sklearn' is not defined...非常感谢任何帮助!
# ElasticNet Regression
from sklearn import linear_model
import statsmodels.api as sm
ElasticNet = sklearn.linear_model.ElasticNet() # create a lasso instance
ElasticNet.fit(X_train, y_train) # fit data
# print(lasso.coef_)
# print (lasso.intercept_) # print out the coefficients
print ("R^2 for training set:"),
print (ElasticNet.score(X_train, y_train))
print ('-'*50)
print ("R^2 for test set:"),
print (ElasticNet.score(X_test, y_test))
【问题讨论】:
-
使用
linear_model.ElasticNet()。删除sklearn
标签: python scikit-learn regression