Closed vincentarelbundock closed 6 months ago
import polars as pl import numpy as np import statsmodels.formula.api as smf from marginaleffects import * mtcars = pl.read_csv("https://vincentarelbundock.github.io/Rdatasets/csv/datasets/mtcars.csv") mod = smf.ols("mpg ~ hp * wt * am", data = mtcars).fit() cond = { "hp": None, "wt": [mtcars["wt"].mean() - mtcars["wt"].std(), mtcars["wt"].mean(), mtcars["wt"].mean() + mtcars["wt"].std()], "am": None } p = plot_predictions(mod, condition = cond)