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()