Quantcast
Channel: Data Science, Analytics and Big Data discussions - Latest topics
Viewing all articles
Browse latest Browse all 4448

Machine learning:-stratified K fold cross_validation

$
0
0

@Chandra_prashad_mish wrote:

can anyone help me to find the error in below code:- **strong text
from sklearn.model_selection import StratifiedKFold
kfold = StratifiedKFold(y = y_train,n_folds = 5, random_state = 2017)

the error i am getting is:-
TypeError Traceback (most recent call last)
in ()
1 from sklearn.model_selection import StratifiedKFold
----> 2 kfold = StratifiedKFold(y = y_train,n_folds = 5, random_state = 2017)

TypeError: init() got an unexpected keyword argument ‘n_folds’

Posts: 1

Participants: 1

Read full topic


Viewing all articles
Browse latest Browse all 4448

Trending Articles