c6ee8ee45b7b854fa2731b6f238e223ec24e1cab,yellowbrick/features/manifold.py,Manifold,fit_transform,#Manifold#Any#Any#,293

Before Change

        self._determine_target_color_type(y)

        // Compute classes and colors if target type is discrete
        if self._target_color_type == DISCRETE:
            self.classes_ = np.unique(y)

            color_kwargs = {"n_colors": len(self.classes_)}

            if isinstance(self.colors, str):
                color_kwargs["colormap"] = self.colors
            else:
                color_kwargs["colors"] = self.colors

            self._colors = resolve_colors(**color_kwargs)

        // Compute target range if colors are continuous
        elif self._target_color_type == CONTINUOUS:
            y = np.asarray(y)
            self.range_ = (y.min(), y.max())

        with Timer() as self.fit_time_:
            Xp = self.manifold.fit_transform(X)

        self.draw(Xp, y)

After Change

        // fit and transform methods, but must do it manually here.

        // Call super fit to compute features, classes, colors, etc.
        super(Manifold, self).fit(X, y)
        with Timer() as self.fit_time_:
            Xp = self.manifold.fit_transform(X)
        self.draw(Xp, y)

Instances