leeper / margins

An R Port of Stata's 'margins' Command
https://cloud.r-project.org/package=margins
Other
260 stars 39 forks source link

Retain model metadata (N, log.lik, R2, etc) #158

Open grantmcdermott opened 3 years ago

grantmcdermott commented 3 years ago

Please specify whether your issue is about:

Background

In https://github.com/tidymodels/broom/pull/909, I bundled some broom tidiers for margins objects. The tidy.margins method is straightforward. However, a glance.margins method proves more difficult, because none of the original model metadata that glance() requires are retained in the margins objects. My hackish solution involves accessing the original model call from the margins object attributes, rerunning this original model, and then calling glance on said original model. This is clearly inefficient and becomes increasingly expensive as we start dealing with large models.

As an aside: A glance.margins method is required by modern table generating packages like modelsummary and huxtable. So we need to find a way to include it if we want to produce nice regression tables with marginal effects. See here.

Possible solutions

I think there are two possible solutions here. They are not mutually exclusive.

  1. Add an option for saving the original model object (perhaps as an attribute) as part of the margins return object.
  2. Include a minimal set of model meta data tags (again, perhaps as attributes) as part of the margins return object, regardless of whether the original model object was retained above or not.

FWIW, we recently faced the same problem with lmtest::coeftest(). Achim implemented both of the above solutions and so I was able to bundle a glance.coeftestmethod that takes advantage of that. Result: More nice regression tables!

Thoughts?

leeper commented 3 years ago

I think we can just add the whole model object as an attribute. Want to send a PR?