e87e84a21e0dfc1258b2f868532b03fabfd66a85,autokeras/adapters/output_adapter.py,ClassificationHeadAdapter,fit_before_convert,#ClassificationHeadAdapter#Any#,67

Before Change

        // If in tf.data.Dataset, must be encoded already.
        if isinstance(dataset, tf.data.Dataset):
            if not self.num_classes:
                shape = data_utils.dataset_shape(dataset)[0]
                // Single column with 0s and 1s.
                if shape == 1:
                    self.num_classes = 2
                else:
                    self.num_classes = shape
            return
        if isinstance(dataset, pd.DataFrame):
            dataset = dataset.values
        if isinstance(dataset, pd.Series):

After Change

        // If encoded.
        if len(dataset.flatten()) != len(dataset):
            if self.num_classes:
                self._check_data_shape(dataset.shape[1:])
            return

        // Fit encoder.
        labels = set(dataset.flatten())

Instances