vincentarelbundock / pymarginaleffects

GNU General Public License v3.0
49 stars 9 forks source link

Dodge linerange when they overlap #63

Closed vincentarelbundock closed 9 months ago

vincentarelbundock commented 9 months ago
import numpy as np
import polars as pl
import statsmodels.formula.api as smf
from marginaleffects import *
from marginaleffects.classes import MarginaleffectsDataFrame
from .utilities import *

dat = pl.read_csv("tests/data/impartiality.csv") \
  .with_columns(pl.col("impartial").cast(pl.Int8))

m = smf.logit(
  "impartial ~ equal * democracy + continent",
  data = dat.to_pandas()
).fit()

p = plot_predictions(m, by = ["democracy", "continent"])