def plot_qq(model: SklearnModel, ax=None) -> None: if ax is None: fig, ax = plt.subplots(1, 1) residuals = model.residuals(model.data.X) sm.qqplot(residuals, fit=True, line="45", ax=ax) ax.set_title("QQ plot") return ax
def plot_homoskedasity_diagnostics(model: SklearnModel, ax=None): 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) -> None: residuals = model.residuals() fig = sm.qqplot(residuals, fit=True, line="45") plt.show()