我尝试使用spark_sklearn.GridSearchCV，但出现init参数错误。
TypeError: __init__() takes at least 4 arguments (4 given)
这是代码:
from spark_sklearn import GridSearchCV
gsearch2 = GridSearchCV(estimator=ensemble.GradientBoostingRegressor(**params), param_grid=param_test2, n_jobs=1)
如果我给GridSearchCV更多的参数，比如添加cv=5，那么错误就会变成
TypeError: __init__() takes at least 4 arguments (5 given)
有什么建议吗？
谢谢。
最佳答案
GridSearchCV.__init__ 需要 3 obligatory arguments :
sc-SparkContext。估计器param_grid。
您忘记了SparkContext:
GridSearchCV(
sc=SparkContext.getOrCreate(),
estimator=ensemble.GradientBoostingRegressor(**params),
param_grid=param_test2, n_jobs=1)
关于python - Spark_sklearn GridSearchCV __init__ 因参数错误而失败，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/47723547/