432331c2ea292f2f8b756901ea3c8bda04817ec5,autosklearn/estimators.py,AutoMLClassifier,_process_target_classes,#AutoMLClassifier#Any#,567

Before Change

    def _process_target_classes(self, y):
        y = np.atleast_1d(y)
        if y.ndim == 2 and y.shape[1] == 1:
            warnings.warn("A column-vector y was passed when a 1d array was"
                          " expected. Please change the shape of y to "
                          "(n_samples,), for example using ravel().",
                          sklearn.utils.DataConversionWarning, stacklevel=2)

        if y.ndim == 1:
            // reshape is necessary to preserve the data contiguity against vs
            // [:, np.newaxis] that does not.
            y = np.reshape(y, (-1, 1))

After Change

        _classes = []
        _n_classes = []

        if self._n_outputs == 1:
            classes_k, y = np.unique(y, return_inverse=True)
            _classes.append(classes_k)
            _n_classes.append(classes_k.shape[0])
        else:
            for k in range(self._n_outputs):
                classes_k, y[:, k] = np.unique(y[:, k], return_inverse=True)
                _classes.append(classes_k)
                _n_classes.append(classes_k.shape[0])

        self._n_classes = np.array(_n_classes, dtype=np.int)

        return y, _classes, _n_classes

Instances