9d0ed839dc9df4c9afc5966c1e3276c478a58d53,autosklearn/metalearning/metafeatures/metafeatures.py,LandmarkDecisionNodeLearner,_calculate,#LandmarkDecisionNodeLearner#Any#Any#Any#,755

Before Change

    def _calculate(self, X, y, categorical):
        import sklearn.tree

        if len(y.shape) == 1 or y.shape[1] == 1:
            kf = sklearn.model_selection.StratifiedKFold(n_splits=5)
        else:
            kf = sklearn.model_selection.KFold(n_splits=5)

After Change

    def _calculate(self, X, y, categorical):
        import sklearn.tree

        if type(y) in ("binary", "multiclass"):
            kf = sklearn.model_selection.StratifiedKFold(n_splits=5)
        else:
            kf = sklearn.model_selection.KFold(n_splits=5)

Instances