spark_sklearn GridSearchCV __init__がパラメータエラーで失敗しました

Question

spark_sklearn.GridSearchCVを使用しようとしましたが、のパラメータエラーが発生しました。

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) 

私は、このようなcv=5を追加するなどGridSearchCVへのより多くのパラメータを与えた場合は、エラーが

TypeError: __init__() takes at least 4 arguments (5 given) 

任意の提案になりますか？

ありがとうございました。

Answer 1

GridSearchCV.__init__は3 obligatory argumentsとります

• scからSparkContext。
• estimator
• param_grid。

あなたはSparkContextを忘れてしまった：

GridSearchCV(
    sc=SparkContext.getOrCreate(), 
    estimator=ensemble.GradientBoostingRegressor(**params), 
    param_grid=param_test2, n_jobs=1)