def process_y_true()

in bindings/jupyter/mlvis/widget_ext/Manifold.py [0:0]

• 7 lines of code
• 3 McCabe index (conditional complexity)

    def process_y_true(self, y_true):
        """
        Convert y true feature into the format manifold recognizes
        :param y_true: An ndarray/list for the ground truth
        :return: A list of the ground truths with the manifold data format
        """
        if isinstance(y_true, pd.DataFrame):
            return y_true.iloc[:, 0].tolist()
        elif isinstance(y_true, np.ndarray):
            return y_true.tolist()
        else:
            return y_true