Python Scikit-learn sklearn.model_selection API

API Reference — scikit-learn 0.23.1 documentation

 

Scikit-learn API - sklearn.model_selection

훈련세트와 테스트 세트를 나눌 때, 교차검증을 위한 검증 세트를 나눌 때, 하이퍼 파라미터 튜닝을 최적화 하기 위한 용도로 사용되는 클래스와 함수들을 모아둔 API

 

• Splitter Classes
  • model_selection.GroupKFold([n_splits])
  • model_selection.GroupShuffleSplit([...])
  • model_selection.KFold([n_splits, shuffle, ...]) 
  • model_selection.LeaveOneGroupOut
  • model_selection.LeavePGroupsOut(n_groups)
  • model_selection.LeaveOneOut
  • model_selection.LeavePOut(p)
  • model_selection.PredefinedSplit(test_fold)
  • model_selection.RepeatedKFold(*[, n_splits, ...])
  • model_selection.RepeatedStratifiedKFold(*[, ...])
  • model_selection.ShuffleSplit([n_splits, ...])
  • model_selection.StratifiedKFold([n_splits, ...])
  • model_selection.StratifiedShuffleSplit([...])
  • model_selection.TimeSeriesSplit([n_splits, ...])
• Splitter Functions
  • model_selection.check_cv([cv, y, classifier])
  • model_selection.train_test_split(\*arrays, ...)
• Hyper-parameter optimizers
  • model_selection.GridSearchCV(estimator, ...)
  • model_selection.ParameterGrid(param_grid)
  • model_selection.ParameterSampler(...[, ...])
  • model_selection.RandomizedSearchCV(...[, ...])
• Model validation
  • model_selection.cross_validate(estimator, X)
  • model_selection.cross_val_predict(estimator, X)
  • model_selection.cross_val_score(estimator, X)
  • model_selection.learning_curve(estimator, X, ...)
  • model_selection.permutation_test_score(...)
  • model_selection.validation_curve (estimator, ...)