Shubham, the definition of ShuffleSplit.__init__ is ShuffleSplit(n_splits=10, test_size=0.1, train_size=None, random_state=None) you are passing the n_split parameter twice (once named and once as the first parameter), as the exception that you getting says, -- Roman On 07/03/17 14:24, Shubham Singh Tomar wrote:
Hi,
I'm trying to use GridSearchCV to tune the parameters for DecisionTreeRegressor. I'm using sklearn 0.18.1
I'm getting the following error:
--------------------------------------------------------------------------- TypeError Traceback (most recent call last) <ipython-input-36-192f7c286a58> in <module>() 1 # Fit the training data to the model using grid search----> 2reg = fit_model(X_train, y_train)3 4 # Produce the value for 'max_depth'5 print "Parameter 'max_depth' is {} for the optimal model.".format(reg.get_params()['max_depth'])<ipython-input-35-500141c331d9> in fit_model(X, y) 11 12 # Create cross-validation sets from the training data---> 13cv_sets = ShuffleSplit(X.shape[0], n_splits = 10, test_size = 0.20, random_state = 0)14 15 # TODO: Create a decision tree regressor objectTypeError: __init__() got multiple values for keyword argument 'n_splits'
-- *Thanks,* *Shubham Singh Tomar* *Autodidact24.github.io <http://Autodidact24.github.io>*
_______________________________________________ scikit-learn mailing list scikit-learn@python.org https://mail.python.org/mailman/listinfo/scikit-learn