search

casper-hansen / Nested-Cross-Validation

Nested cross-validation for unbiased predictions. Can be used with Scikit-Learn, XGBoost, Keras and LightGBM, or any other estimator that implements the scikit-learn interface.
MIT License
62 stars 20 forks source link

Add random_state and allow using StratifiedKFold #11

Closed iajst17 closed 4 years ago

iajst17 commented 4 years ago

Hi Casper,

Thank you so much for your program. I suggest to add random_state to reproduce the results and allow StratifiedKFold option as well.

Kind regards,

Ken

casper-hansen commented 4 years ago

This issue is linked to #7

I'm working on it now, but it might not be released until there are more features added

casper-hansen commented 4 years ago

This is now a feature.

But it is not released, so pull the github package with pip like this:

pip install git+https://github.com/casperbh96/Nested-Cross-Validation.git

iajst17 commented 4 years ago

Thank you. Hope to see your new version of NestedCV soon.