Functions

def plot_homoscedasticity_diagnostics(model: SklearnModel, ax=None)

Expand source code

def plot_homoscedasticity_diagnostics(model: SklearnModel, ax=None):
    require_optional_dependency('seaborn', 'plot_homoscedasticity_diagnostics',
                                purpose='draws the regression plot')
    import seaborn as sns

    if ax is None:
        _, ax = plt.subplots(1, 1, figsize=(5, 5))
    sns.regplot(model.predict(model.data.X.values), model.residuals(model.data.X.values), ax=ax)
    ax.set_title("Fitted Values V Residuals")
    ax.set_xlabel("Fitted Value")
    ax.set_ylabel("Residual")
    return ax

def plot_qq(model: SklearnModel, ax=None) ‑> None

Expand source code

def plot_qq(model: SklearnModel, ax=None) -> None:
    require_optional_dependency('statsmodels', 'plot_qq',
                                purpose='draws the QQ plot')
    import statsmodels.api as sm

    if ax is None:
        _, ax = plt.subplots(1, 1)
    residuals = model.residuals(model.data.X.values)
    sm.qqplot(residuals, fit=True, line="45", ax=ax)
    ax.set_title("QQ plot")
    return ax