rstanarm-datasets: Datasets for rstanarm examples

Description Format References Examples


Small datasets for use in rstanarm examples and vignettes.



Data on hits and at-bats from the 1970 Major League Baseball season for 18 players.

Source: Efron and Morris (1975).

18 obs. of 5 variables

  • Player Player's last name

  • Hits Number of hits in the first 45 at-bats of the season

  • AB Number of at-bats (45 for all players)

  • RemainingAB Number of remaining at-bats (different for most players)

  • RemainingHits Number of remaining hits


Hits and at-bats for the entire 2006 American League season of Major League Baseball.

Source: Carpenter (2009)

302 obs. of 2 variables

  • y Number of hits

  • K Number of at-bats


Data from a survey of adult American women and their children (a subsample from the National Longitudinal Survey of Youth).

Source: Gelman and Hill (2007)

434 obs. of 4 variables

  • kid_score Child's IQ score

  • mom_hs Indicator for whether the mother has a high school degree

  • mom_iq Mother's IQ score

  • mom_age Mother's age


Surgical mortality rates in 12 hospitals performing cardiac surgery in babies.

Source: Spiegelhalter et al. (1996).

12 obs. of 2 variables

  • y Number of deaths

  • K Number of surgeries


Longitudinal biomarker and time-to-event survival data for 40 patients with primary biliary cirrhosis who participated in a randomised placebo controlled trial of D-penicillamine conducted at the Mayo Clinic between 1974 and 1984.

Source: Therneau and Grambsch (2000)

304 obs. of 8 variables (pbcLong) and 40 obs. of 7 variables (pbcSurv)

  • age in years

  • albumin serum albumin (g/dl)

  • logBili logarithm of serum bilirubin

  • death indicator of death at endpoint

  • futimeYears time (in years) between baseline and the earliest of death, transplantion or censoring

  • id numeric ID unique to each individual

  • platelet platelet count

  • sex gender (m = male, f = female)

  • status status at endpoint (0 = censored, 1 = transplant, 2 = dead)

  • trt binary treatment code (0 = placebo, 1 = D-penicillamine)

  • year time (in years) of the longitudinal measurements, taken as time since baseline)


Data on radon levels in houses in the state of Minnesota.

Source: Gelman and Hill (2007)

919 obs. of 4 variables

  • log_radon Radon measurement from the house (log scale)

  • log_uranium Uranium level in the county (log scale)

  • floor Indicator for radon measurement made on the first floor of the house (0 = basement, 1 = first floor)

  • county County name (factor)


Data on the efficacy of a pest management system at reducing the number of roaches in urban apartments.

Source: Gelman and Hill (2007)

262 obs. of 6 variables

  • y Number of roaches caught

  • roach1 Pretreatment number of roaches

  • treatment Treatment indicator

  • senior Indicator for only eldery residents in building

  • exposure2 Number of days for which the roach traps were used


Tarone (1982) provides a data set of tumor incidence in historical control groups of rats; specifically endometrial stromal polyps in female lab rats of type F344.

Source: Gelman and Hill (2007)

71 obs. of 2 variables

  • y Number of rats with tumors

  • K Number of rats


A survey of 3200 residents in a small area of Bangladesh suffering from arsenic contamination of groundwater. Respondents with elevated arsenic levels in their wells had been encouraged to switch their water source to a safe public or private well in the nearby area and the survey was conducted several years later to learn which of the affected residents had switched wells.

Souce: Gelman and Hill (2007)

3020 obs. of 5 variables

  • switch Indicator for well-switching

  • arsenic Arsenic level in respondent's well

  • dist Distance (meters) from the respondent's house to the nearest well with safe drinking water.

  • association Indicator for member(s) of household participate in community organizations

  • educ Years of education (head of household)


Carpenter, B. (2009) Bayesian estimators for the beta-binomial model of batting ability.

Efron, B. and Morris, C. (1975) Data analysis using Stein's estimator and its generalizations. Journal of the American Statistical Association 70(350), 311–319.

Gelman, A. and Hill, J. (2007). Data Analysis Using Regression and Multilevel/Hierarchical Models. Cambridge University Press, Cambridge, UK.

Spiegelhalter, D., Thomas, A., Best, N., & Gilks, W. (1996) BUGS 0.5 Examples. MRC Biostatistics Unit, Institute of Public health, Cambridge, UK.

Tarone, R. E. (1982) The use of historical control information in testing for a trend in proportions. Biometrics 38(1):215–220.

Therneau, T. and Grambsch, P. (2000) Modeling Survival Data: Extending the Cox Model. Springer-Verlag, New York, US.


# Using 'kidiq' dataset 
fit <- stan_lm(kid_score ~ mom_hs * mom_iq, data = kidiq, 
               prior = R2(location = 0.30, what = "mean"),
               # the next line is only to make the example go fast enough
               chains = 1, iter = 500, seed = 12345)
pp_check(fit, nreps = 20)

pp_check(fit, plotfun = "stat_grouped", stat = "median", 
         group = factor(kidiq$mom_hs, labels = c("No HS", "HS")))

rstanarm documentation built on Oct. 4, 2019, 1:04 a.m.