forked from tidymodels/broom
-
Notifications
You must be signed in to change notification settings - Fork 1
/
README.Rmd
278 lines (192 loc) · 12.3 KB
/
README.Rmd
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
broom: let's tidy up a bit
=====================
```{r echo = FALSE}
knitr::opts_chunk$set(warning = FALSE, message = FALSE)
```
[![CRAN_Status_Badge](http://www.r-pkg.org/badges/version/broom)](https://CRAN.R-project.org/package=broom)
[![Travis-CI Build Status](https://travis-ci.org/tidyverse/broom.svg?branch=master)](https://travis-ci.org/tidyverse/broom)
[![AppVeyor Build Status](https://ci.appveyor.com/api/projects/status/github/tidyverse/broom?branch=master&svg=true)](https://ci.appveyor.com/project/tidyverse/broom)
[![Coverage Status](https://img.shields.io/codecov/c/github/tidyverse/broom/master.svg)](https://codecov.io/github/tidyverse/broom?branch=master)
The broom package takes the messy output of built-in functions in R, such as `lm`, `nls`, or `t.test`, and turns them into tidy data frames.
The concept of "tidy data", [as introduced by Hadley Wickham](http://www.jstatsoft.org/v59/i10), offers a powerful framework for data manipulation and analysis. That paper makes a convincing statement of the problem this package tries to solve (emphasis mine):
> **While model inputs usually require tidy inputs, such attention to detail doesn't carry over to model outputs. Outputs such as predictions and estimated coefficients aren't always tidy. This makes it more difficult to combine results from multiple models.** For example, in R, the default representation of model coefficients is not tidy because it does not have an explicit variable that records the variable name for each estimate, they are instead recorded as row names. In R, row names must be unique, so combining coefficients from many models (e.g., from bootstrap resamples, or subgroups) requires workarounds to avoid losing important information. **This knocks you out of the flow of analysis and makes it harder to combine the results from multiple models. I'm not currently aware of any packages that resolve this problem.**
broom is an attempt to bridge the gap from untidy outputs of predictions and estimations to the tidy data we want to work with. It centers around three S3 methods, each of which take common objects produced by R statistical functions (`lm`, `t.test`, `nls`, etc) and convert them into a data frame. broom is particularly designed to work with Hadley's [dplyr](https://github.com/hadley/dplyr) package (see the "broom and dplyr" vignette for more).
broom should be distinguished from packages like [reshape2](https://CRAN.R-project.org/package=reshape2) and [tidyr](https://CRAN.R-project.org/package=tidyr), which rearrange and reshape data frames into different forms. Those packages perform critical tasks in tidy data analysis but focus on manipulating data frames in one specific format into another. In contrast, broom is designed to take format that is *not* in a data frame (sometimes not anywhere close) and convert it to a tidy data frame.
Tidying model outputs is not an exact science, and it's based on a judgment of the kinds of values a data scientist typically wants out of a tidy analysis (for instance, estimates, test statistics, and p-values). You may lose some of the information in the original object that you wanted, or keep more information than you need. If you think the tidy output for a model should be changed, or if you're missing a tidying function for an S3 class that you'd like, I strongly encourage you to [open an issue](http://github.com/tidyverse/broom/issues) or a pull request.
Installation and Documentation
------------
The broom package is available on CRAN:
install.packages("broom")
You can also install the development version of the broom package using [devtools](https://github.com/hadley/devtools):
```
library(devtools)
install_github("tidyverse/broom")
```
For additional documentation, please browse the vignettes:
```
browseVignettes(package="broom")
```
Tidying functions
-----------------
This package provides three S3 methods that do three distinct kinds of tidying.
* `tidy`: constructs a data frame that summarizes the model's statistical findings. This includes coefficients and p-values for each term in a regression, per-cluster information in clustering applications, or per-test information for `multtest` functions.
* `augment`: add columns to the original data that was modeled. This includes predictions, residuals, and cluster assignments.
* `glance`: construct a concise *one-row* summary of the model. This typically contains values such as R^2, adjusted R^2, and residual standard error that are computed once for the entire model.
Note that some classes may have only one or two of these methods defined.
Consider as an illustrative example a linear fit on the built-in `mtcars` dataset.
```{r}
lmfit <- lm(mpg ~ wt, mtcars)
lmfit
```
```{r}
summary(lmfit)
```
This summary output is useful enough if you just want to read it. However, converting it to a data frame that contains all the same information, so that you can combine it with other models or do further analysis, is not trivial. You have to do `coef(summary(lmfit))` to get a matrix of coefficients, the terms are still stored in row names, and the column names are inconsistent with other packages (e.g. `Pr(>|t|)` compared to `p.value`).
Instead, you can use the `tidy` function, from the broom package, on the fit:
```{r}
library(broom)
tidy(lmfit)
```
This gives you a data.frame representation. Note that the row names have been moved into a column called `term`, and the column names are simple and consistent (and can be accessed using `$`).
Instead of viewing the coefficients, you might be interested in the fitted values and residuals for each of the original points in the regression. For this, use `augment`, which augments the original data with information from the model:
```{r}
head(augment(lmfit))
```
Note that each of the new columns begins with a `.` (to avoid overwriting any of the original columns).
Finally, several summary statistics are computed for the entire regression, such as R^2 and the F-statistic. These can be accessed with the `glance` function:
```{r}
glance(lmfit)
```
This distinction between the `tidy`, `augment` and `glance` functions is explored in a different context in the k-means vignette.
Other Examples
--------------
### Generalized linear and non-linear models
These functions apply equally well to the output from `glm`:
```{r}
glmfit <- glm(am ~ wt, mtcars, family="binomial")
tidy(glmfit)
```
```{r}
head(augment(glmfit))
```
```{r}
glance(glmfit)
```
Note that the statistics computed by `glance` are different for `glm` objects than for `lm` (e.g. deviance rather than R^2):
These functions also work on other fits, such as nonlinear models (`nls`):
```{r}
nlsfit <- nls(mpg ~ k / wt + b, mtcars, start=list(k=1, b=0))
tidy(nlsfit)
```
```{r}
head(augment(nlsfit, mtcars))
```
```{r}
glance(nlsfit)
```
### Hypothesis testing
The `tidy` function can also be applied to `htest` objects, such as those output by popular built-in functions like `t.test`, `cor.test`, and `wilcox.test`.
```{r}
tt <- t.test(wt ~ am, mtcars)
tidy(tt)
```
Some cases might have fewer columns (for example, no confidence interval):
```{r warning = FALSE}
wt <- wilcox.test(wt ~ am, mtcars)
tidy(wt)
```
Since the `tidy` output is already only one row, `glance` returns the same output:
```{r}
glance(tt)
```
```{r}
glance(wt)
```
There is no `augment` function for `htest` objects, since there is no meaningful sense in which a hypothesis test produces output about each initial data point.
### Available Tidiers
Currently broom provides tidying methods for many S3 objects from the built-in stats package, including
* `lm`
* `glm`
* `htest`
* `anova`
* `nls`
* `kmeans`
* `manova`
* `TukeyHSD`
* `arima`
It also provides methods for S3 objects in popular third-party packages, including
* `lme4`
* `glmnet`
* `boot`
* `gam`
* `survival`
* `lfe`
* `zoo`
* `multcomp`
* `sp`
* `maps`
A full list of the `tidy`, `augment` and `glance` methods available for each class is as follows:
```{R, echo = FALSE, message = FALSE}
library(broom)
library(dplyr)
tidy_methods <- methods("tidy")
glance_methods <- methods("glance")
augment_methods <- methods("augment")
tidy_methods <- data.frame(class = sub("tidy[.]", "",
as.vector(tidy_methods)),
tidy = "x",
stringsAsFactors=FALSE)
glance_methods <- data.frame(class = sub("glance[.]", "",
as.vector(glance_methods)),
glance = "x",
stringsAsFactors = FALSE)
augment_methods <- data.frame(class = sub("augment[.]", "",
as.vector(augment_methods)),
augment = "x",
stringsAsFactors = FALSE)
TidyMethods <- left_join(tidy_methods, glance_methods) %>%
left_join(augment_methods) %>%
mutate(tidy = ifelse(is.na(tidy), "", tidy),
glance = ifelse(is.na(glance), "", glance),
augment = ifelse(is.na(augment), "", augment))
names(TidyMethods) <- c("Class", "`tidy`", "`glance`", "`augment`")
knitr::kable(TidyMethods)
```
Conventions
------------
In order to maintain consistency, we attempt to follow some conventions regarding the structure of returned data.
### All functions
* The output of the `tidy`, `augment` and `glance` functions is *always* a data frame.
* The output never has rownames. This ensures that you can combine it with other tidy outputs without fear of losing information (since rownames in R cannot contain duplicates).
* Some column names are kept consistent, so that they can be combined across different models and so that you know what to expect (in contrast to asking "is it `pval` or `PValue`?" every time). The examples below are not all the possible column names, nor will all tidy output contain all or even any of these columns.
### tidy functions
* Each row in a `tidy` output typically represents some well-defined concept, such as one term in a regression, one test, or one cluster/class. This meaning varies across models but is usually self-evident. The one thing each row cannot represent is a point in the initial data (for that, use the `augment` method).
* Common column names include:
* `term`: the term in a regression or model that is being estimated.
* `p.value`: this spelling was chosen (over common alternatives such as `pvalue`, `PValue`, or `pval`) to be consistent with functions in R's built-in `stats` package
* `statistic` a test statistic, usually the one used to compute the p-value. Combining these across many sub-groups is a reliable way to perform (e.g.) bootstrap hypothesis testing
* `estimate` estimate of an effect size, slope, or other value
* `std.error` standard error
* `conf.low` the low end of a confidence interval on the `estimate`
* `conf.high` the high end of a confidence interval on the `estimate`
* `df` degrees of freedom
### augment functions
* `augment(model, data)` adds columns to the original data.
* If the `data` argument is missing, `augment` attempts to reconstruct the data from the model (note that this may not always be possible, and usually won't contain columns not used in the model).
* Each row in an `augment` output matches the corresponding row in the original data.
* If the original data contained rownames, `augment` turns them into a column called `.rownames`.
* Newly added column names begin with `.` to avoid overwriting columns in the original data.
* Common column names include:
* `.fitted`: the predicted values, on the same scale as the data.
* `.resid`: residuals: the actual y values minus the fitted values
* `.cluster`: cluster assignments
### glance functions
* `glance` always returns a one-row data frame.
* The only exception is that `glance(NULL)` returns an empty data frame.
* We avoid including arguments that were *given* to the modeling function. For example, a `glm` glance output does not need to contain a field for `family`, since that is decided by the user calling `glm` rather than the modeling function itself.
* Common column names include:
* `r.squared` the fraction of variance explained by the model
* `adj.r.squared` R^2 adjusted based on the degrees of freedom
* `sigma` the square root of the estimated variance of the residuals
### Code of Conduct
Please note that this project is released with a [Contributor Code of Conduct](CONDUCT.md). By participating in this project you agree to abide by its terms.