c68df783deee8bf210f7e33f80f9ab2c46d3108c,yellowbrick/classifier/class_prediction_error.py,,class_prediction_error,#,241

Before Change

    )

    // Create the train and test splits
    X_train, X_test, y_train, y_test = tts(
        X, y, test_size=test_size, random_state=random_state
    )

    // Fit and transform the visualizer (calls draw)
    visualizer.fit(X_train, y_train, **kwargs)
    visualizer.score(X_test, y_test)
    visualizer.finalize()

    // Return the visualizer
    return visualizer

After Change

    viz.fit(X_train, y_train, **kwargs)

    // Score the visualizer
    if X_test is not None and y_test is not None:
        viz.score(X_test, y_test)
    elif X_test is not None or y_test is not None:
        raise YellowbrickValueError("must specify both X_test and y_test or neither")
    else:
        viz.score(X_train, y_train)

    // Draw the final visualization
    if show:
        viz.show()
    else:
        viz.finalize()

Instances