vincentarelbundock / pymarginaleffects

GNU General Public License v3.0
47 stars 8 forks source link

Get started bug #62

Closed vincentarelbundock closed 6 months ago

vincentarelbundock commented 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)