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

Before Change

    visualizer = PredictionError(model, ax, alpha=alpha, is_fitted=is_fitted, **kwargs)

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

    // 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

    visualizer.fit(X_train, y_train)

    // 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