a0e281ec3bdf753aa025b38547b829f145797373,yellowbrick/regressor/residuals.py,,prediction_error,#,288

Before Change

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

    // Return the axes object on the visualizer
    return visualizer

After Change

    // Scores the visualizer with X and y test if provided, X and y train if not
    if X_test is not None and y_test is not None:
        visualizer.score(X_test, y_test)
    elif X_test is not None or y_test is not None:
        raise YellowbrickValueError(
            "both X_test and y_test are required if one is specified"
        )
    else:
        visualizer.score(X_train, y_train)

    if show:
        visualizer.show()
    else:
        visualizer.finalize()

    // Return the axes object on the visualizer
    return visualizer


////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

Instances