# 18Simple regression

library(tidyverse)

## 18.1 Rainfall in California

The data in link are rainfall and other measurements for 30 weather stations in California. Our aim is to understand how well the annual rainfall at these stations (measured in inches) can be predicted from the other measurements, which are the altitude (in feet above sea level), the latitude (degrees north of the equator) and the distance from the coast (in miles).

1. Read the data into R. You’ll have to be careful here, since the values are space-delimited, but sometimes by more than one space, to make the columns line up. read_table2, with filename or url, will read it in. One of the variables is called rainfall, so as long as you do not call the data frame that, you should be safe.

2. Make a boxplot of the rainfall figures, and explain why the values are reasonable. (A rainfall cannot be negative, and it is unusual for a annual rainfall to exceed 60 inches.) A ggplot boxplot needs something on the $$x$$-axis: the number 1 will do.

3. Plot rainfall against each of the other quantitative variables (that is, not station).

4. Look at the relationship of each other variable with rainfall. Justify the assertion that latitude seems most strongly related with rainfall. Is that relationship positive or negative? linear? Explain briefly.

5. Fit a regression with rainfall as the response variable, and latitude as your explanatory variable. What are the intercept, slope and R-squared values? Is there a significant relationship between rainfall and your explanatory variable? What does that mean?

6. Fit a multiple regression predicting rainfall from all three of the other (quantitative) variables. Display the results. Comment is coming up later.

7. What is the R-squared for the regression of the last part? How does that compare with the R-squared of your regression in part (e)?

8. What do you conclude about the importance of the variables that you did not include in your model in (e)? Explain briefly.

9. Make a suitable hypothesis test that the variables altitude and fromcoast significantly improve the prediction of rainfall over the use of latitude alone. What do you conclude?

## 18.2 Carbon monoxide in cigarettes

The (US) Federal Trade Commission assesses cigarettes according to their tar, nicotine and carbon monoxide contents. In a particular year, 25 brands were assessed. For each brand, the tar, nicotine and carbon monoxide (all in milligrams) were measured, along with the weight in grams. Our aim is to predict carbon monoxide from any or all of the other variables. The data are in link. These are aligned by column (except for the variable names), with more than one space between each column of data.

1. Read the data into R, and check that you have 25 observations and 4 variables.

2. Run a regression to predict carbon monoxide from the other variables, and obtain a summary of the output.

3. Which one of your explanatory variables would you remove from this regression? Explain (very) briefly. Go ahead and fit the regression without it, and describe how the change in R-squared from the regression in (b) was entirely predictable.

4. Fit a regression predicting carbon monoxide from nicotine only, and display the summary.

5. nicotine was far from being significant in the model of (c), and yet in the model of (d), it was strongly significant, and the R-squared value of (d) was almost as high as that of (c). What does this say about the importance of nicotine as an explanatory variable? Explain, as briefly as you can manage.

6. Make a “pairs plot”: that is, scatter plots between all pairs of variables. This can be done by feeding the whole data frame into plot.1 Do you see any strong relationships that do not include co? Does that shed any light on the last part? Explain briefly (or “at length” if that’s how it comes out).

## 18.3 Maximal oxygen uptake in young boys

A physiologist wanted to understand the relationship between physical characteristics of pre-adolescent boys and their maximal oxygen uptake (millilitres of oxygen per kilogram of body weight). The data are in link for a random sample of 10 pre-adolescent boys. The variables are (with units):

• uptake: Oxygen uptake (millitres of oxygen per kilogram of body weight)

• age: boy’s age (years)

• height: boy’s height (cm)

• weight: boy’s weight (kg)

• chest: chest depth (cm).

1. Read the data into R and confirm that you do indeed have 10 observations.

2. Fit a regression predicting oxygen uptake from all the other variables, and display the results.

3. (A one-mark question.) Would you say, on the evidence so far, that the regression fits well or badly? Explain (very) briefly.

4. It seems reasonable that an older boy should have a greater oxygen uptake, all else being equal. Is this supported by your output? Explain briefly.

5. It seems reasonable that a boy with larger weight should have larger lungs and thus a statistically significantly larger oxygen uptake. Is that what happens here? Explain briefly.

6. Fit a model that contains only the significant explanatory variables from your first regression. How do the R-squared values from the two regressions compare? (The last sentence asks for more or less the same thing as the next part. Answer it either here or there. Either place is good.)

7. How has R-squared changed between your two regressions? Describe what you see in a few words.

8. Carry out a test comparing the fit of your two regression models. What do you conclude, and therefore what recommendation would you make about the regression that would be preferred?

9. Obtain a table of correlations between all the variables in the data frame. Do this by feeding the whole data frame into cor. We found that a regression predicting oxygen uptake from just height was acceptably good. What does your table of correlations say about why that is? (Hint: look for all the correlations that are large.)

## 18.4 Facebook friends and grey matter

Is there a relationship between the number of Facebook friends a person has, and the density of grey matter in the areas of the brain associated with social perception and associative memory? To find out, a 2012 study measured both of these variables for a sample of 40 students at City University in London (England). The data are at link. The grey matter density is on a $$z$$-score standardized scale. The values are separated by tabs.

The aim of this question is to produce a Quarto report that contains your answers to the questions below.

You should aim to make your report flow smoothly, so that it would be pleasant for a grader to read, and can stand on its own as an analysis (rather than just being the answer to a question that I set you). Some suggestions: give your report a title and arrange it into sections with an Introduction; add a small amount of additional text here and there explaining what you are doing and why. I don’t expect you to spend a large amount of time on this, but I do hope you will make some effort. (My report came out to 4 Word pages; HTML of course doesn’t have page numbers, but that’s a good kind of length to aim for.)

1. Read in the data and make a scatterplot for predicting the number of Facebook friends from the grey matter density. On your scatterplot, add a smooth trend.

2. Describe what you see on your scatterplot: is there a trend, and if so, what kind of trend is it? (Don’t get too taken in by the exact shape of your smooth trend.) Think “form, direction, strength”.

3. Fit a regression predicting the number of Facebook friends from the grey matter density, and display the output.

4. Is the slope of your regression line significantly different from zero? What does that mean, in the context of the data?

5. Are you surprised by the results of parts (b) and (d)? Explain briefly.

6. Obtain a scatterplot with the regression line on it.

7. Obtain a plot of the residuals from the regression against the fitted values, and comment briefly on it.

## 18.5 Endogenous nitrogen excretion in carp

A paper in Fisheries Science reported on variables that affect “endogenous nitrogen excretion” or ENE in carp raised in Japan. A number of carp were divided into groups based on body weight, and each group was placed in a different tank. The mean body weight of the carp placed in each tank was recorded. The carp were then fed a protein-free diet three times daily for a period of 20 days. At the end of the experiment, the amount of ENE in each tank was measured, in milligrams of total fish body weight per day. (Thus it should not matter that some of the tanks had more fish than others, because the scaling is done properly.)

For this question, write a report in a Quarto document that answers the questions below and contains some narrative that describes your analysis. Create an HTML document from your Quarto document.

2. Create a scatterplot of ENE (response) against bodyweight (explanatory). Add a smooth trend to your plot.

3. Is there an upward or downward trend (or neither)? Is the relationship a line or a curve? Explain briefly.

4. Fit a straight line to the data, and obtain the R-squared for the regression.

5. Obtain a residual plot (residuals against fitted values) for this regression. Do you see any problems? If so, what does that tell you about the relationship in the data?

6. Fit a parabola to the data (that is, including an $$x$$-squared term). Compare the R-squared values for the models in this part and part (d). Does that suggest that the parabola model is an improvement here over the linear model?

7. Is the test for the slope coefficient for the squared term significant? What does this mean?

8. Make the scatterplot of part (b), but add the fitted curve. Describe any way in which the curve fails to fit well.

9. Obtain a residual plot for the parabola model. Do you see any problems with it? (If you do, I’m not asking you to do anything about them in this question, but I will.)

## 18.6 Salaries of social workers

Another salary-prediction question: does the number of years of work experience that a social worker has help to predict their salary? Data for 50 social workers are in link.

1. Read the data into R. Check that you have 50 observations on two variables. Also do something to check that the years of experience and annual salary figures look reasonable overall.

2. Make a scatterplot showing how salary depends on experience. Does the nature of the trend make sense?

3. Fit a regression predicting salary from experience, and display the results. Is the slope positive or negative? Does that make sense?

4. Obtain and plot the residuals against the fitted values. What problem do you see?

5. The problem you unearthed in the previous part is often helped by a transformation. Run Box-Cox on your data to find a suitable transformation. What transformation is suggested?

6. Calculate a new variable as suggested by your transformation. Use your transformed response in a regression, showing the summary.

7. Obtain and plot the residuals against the fitted values for this regression. Do you seem to have solved the problem with the previous residual plot?

## 18.7 Predicting volume of wood in pine trees

In forestry, the financial value of a tree is the volume of wood that it contains. This is difficult to estimate while the tree is still standing, but the diameter is easy to measure with a tape measure (to measure the circumference) and a calculation involving $$\pi$$, assuming that the cross-section of the tree is at least approximately circular. The standard measurement is “diameter at breast height” (that is, at the height of a human breast or chest), defined as being 4.5 feet above the ground.

Several pine trees had their diameter measured shortly before being cut down, and for each tree, the volume of wood was recorded. The data are in link. The diameter is in inches and the volume is in cubic inches. Is it possible to predict the volume of wood from the diameter?

1. Read the data into R and display the values (there are not very many).

2. Make a suitable plot.

3. Describe what you learn from your plot about the relationship between diameter and volume, if anything.

4. Fit a (linear) regression, predicting volume from diameter, and obtain the summary. How would you describe the R-squared?

5. Draw a graph that will help you decide whether you trust the linearity of this regression. What do you conclude? Explain briefly.

6. What would you guess would be the volume of a tree of diameter zero? Is that what the regression predicts? Explain briefly.

7. A simple way of modelling a tree’s shape is to pretend it is a cone, like this, but probably taller and skinnier:

with its base on the ground. What is the relationship between the diameter (at the base) and volume of a cone? (If you don’t remember, look it up. You’ll probably get a formula in terms of the radius, which you’ll have to convert. Cite the website you used.)

1. Fit a regression model that predicts volume from diameter according to the formula you obtained in the previous part. You can assume that the trees in this data set are of similar heights, so that the height can be treated as a constant.
Display the results.

## 18.8 Tortoise shells and eggs

A biologist measured the length of the carapace (shell) of female tortoises, and then x-rayed the tortoises to count how many eggs they were carrying. The length is measured in millimetres. The data are in link. The biologist is wondering what kind of relationship, if any, there is between the carapace length (as an explanatory variable) and the number of eggs (as a response variable).

1. Read in the data, and check that your values look reasonable.

2. Obtain a scatterplot, with a smooth trend, of the data.

3. The biologist expected that a larger tortoise would be able to carry more eggs. Is that what the scatterplot is suggesting? Explain briefly why or why not.

4. Fit a straight-line relationship and display the summary.

5. Add a squared term to your regression, fit that and display the summary.

6. Is a curve better than a line for these data? Justify your answer in two ways: by comparing a measure of fit, and by doing a suitable test of significance.

7. Make a residual plot for the straight line model: that is, plot the residuals against the fitted values. Does this echo your conclusions of the previous part? In what way? Explain briefly.

## 18.9 Roller coasters

A poll on the Discovery Channel asked people to nominate the best roller-coasters in the United States. We will examine the 10 roller-coasters that received the most votes. Two features of a roller-coaster that are of interest are the distance it drops from start to finish, measured here in feet2 and the duration of the ride, measured in seconds. Is it true that roller-coasters with a bigger drop also tend to have a longer ride? The data are at link.3

1. Read the data into R and verify that you have a sensible number of rows and columns.

2. Make a scatterplot of duration (response) against drop (explanatory), labelling each roller-coaster with its name in such a way that the labels do not overlap. Add a regression line to your plot.

3. Would you say that roller-coasters with a larger drop tend to have a longer ride? Explain briefly.

4. Find a roller-coaster that is unusual compared to the others. What about its combination of drop and duration is unusual?

## 18.10 Running and blood sugar

A diabetic wants to know how aerobic exercise affects his blood sugar. When his blood sugar reaches 170 (mg/dl), he goes out for a run at a pace of 10 minutes per mile. He runs different distances on different days. Each time he runs, he measures his blood sugar after the run. (The preferred blood sugar level is between 80 and 120 on this scale.) The data are in the file link. Our aim is to predict blood sugar from distance.

1. Read in the data and display the data frame that you read in.

2. Make a scatterplot and add a smooth trend to it.

3. Would you say that the relationship between blood sugar and running distance is approximately linear, or not? It is therefore reasonable to use a regression of blood sugar on distance? Explain briefly.

4. Fit a suitable regression, and obtain the regression output.

5. How would you interpret the slope? That is, what is the slope, and what does that mean about blood sugar and running distance?

6. Is there a (statistically) significant relationship between running distance and blood sugar? How do you know? Do you find this surprising, given what you have seen so far? Explain briefly.

7. This diabetic is planning to go for a 3-mile run tomorrow and a 5-mile run the day after. Obtain suitable 95% intervals that say what his blood sugar might be after each of these runs.

8. Which of your two intervals is longer? Does this make sense? Explain briefly.

## 18.11 Calories and fat in pizza

The file at link came from a spreadsheet of information about 24 brands of pizza: specifically, per 5-ounce serving, the number of calories, the grams of fat, and the cost (in US dollars). The names of the pizza brands are quite long. This file may open in a spreadsheet when you browse to the link, depending on your computer’s setup.

1. Read in the data and display at least some of the data frame. Are the variables of the right types? (In particular, why is the number of calories labelled one way and the cost labelled a different way?)

2. Make a scatterplot for predicting calories from the number of grams of fat. Add a smooth trend. What kind of relationship do you see, if any?

3. Fit a straight-line relationship, and display the intercept, slope, R-squared, etc. Is there a real relationship between the two variables, or is any apparent trend just chance?

4. Obtain a plot of the residuals against the fitted values for this regression. Does this indicate that there are any problems with this regression, or not? Explain briefly.

5. The research assistant in this study returns with two new brands of pizza (ones that were not in the original data). The fat content of a 5-ounce serving was 12 grams for the first brand and 20 grams for the second brand. For each of these brands of pizza, obtain a suitable 95% interval for the number of calories contained in a 5-ounce serving.

## 18.12 Where should the fire stations be?

In city planning, one major issue is where to locate fire stations. If a city has too many fire stations, it will spend too much on running them, but if it has too few, there may be unnecessary fire damage because the fire trucks take too long to get to the fire.

The first part of a study of this kind of issue is to understand the relationship between the distance from the fire station (measured in miles in our data set) and the amount of fire damage caused (measured in thousands of dollars). A city recorded the fire damage and distance from fire station for 15 residential fires (which you can take as a sample of “all possible residential fires in that city”). The data are in link.

1. Read in and display the data, verifying that you have the right number of rows and the right columns.

2. * Obtain a 95% confidence interval for the mean fire damage. (There is nothing here from STAD29, and your answer should have nothing to do with distance.)

3. Draw a scatterplot for predicting the amount of fire damage from the distance from the fire station. Add a smooth trend to your plot.

4. * Is there a relationship between distance from fire station and fire damage? Is it linear or definitely curved? How strong is it? Explain briefly.

5. Fit a regression predicting fire damage from distance. How is the R-squared consistent (or inconsistent) with your answer from part~(here)?

6. Obtain a 95% confidence interval for the mean fire damage for a residence that is 4 miles from the nearest fire station*. (Note the contrast with part~(here).)

7. Compare the confidence intervals of parts (here) and (here). Specifically, compare their centres and their lengths, and explain briefly why the results make sense.

## 18.13 Making it stop

If you are driving, and you hit the brakes, how far do you travel before coming to a complete stop? Presumably this depends on how fast you are going. Knowing this relationship is important in setting speed limits on roads. For example, on a very bendy road, the speed limit needs to be low, because you cannot see very far ahead, and there could be something just out of sight that you need to stop for.

Data were collected for a typical car and driver, as shown in http://ritsokiguess.site/datafiles/stopping.csv. These are American data, so the speeds are miles per hour and the stopping distances are in feet.

1. Read in and display (probably all of) the data.

2. Make a suitable plot of the data.

3. Describe any trend you see in your graph.

4. Fit a linear regression predicting stopping distance from speed. (You might have some misgivings about doing this, but do it anyway.)

5. Plot the residuals against the fitted values for this regression.

6. What do you learn from the residual plot? Does that surprise you? Explain briefly.

7. What is the actual relationship between stopping distance and speed, according to the physics? See if you can find out. Cite any books or websites that you use: that is, include a link to a website, or give enough information about a book that the grader could find it.

8. Fit the relationship that your research indicated (in the previous part) and display the results. Comment briefly on the R-squared value.

9. Somebody says to you “if you have a regression with a high R-squared, like 95%, there is no need to look for a better model.” How do you respond to this? Explain briefly.

## 18.14 Predicting height from foot length

Is it possible to estimate the height of a person from the length of their foot? To find out, 33 (male) students had their height and foot length measured. The data are in http://ritsokiguess.site/datafiles/heightfoot.csv.

1. Read in and display (some of) the data. (If you are having trouble, make sure you have exactly the right URL. The correct URL has no spaces or other strange characters in it.)

2. Make a suitable plot of the two variables in the data frame.

3. Are there any observations not on the trend of the other points? What is unusual about those observations?

4. Fit a regression predicting height from foot length, including any observations that you identified in the previous part. For that regression, plot the residuals against the fitted values and make a normal quantile plot of the residuals.

5. Earlier, you identified one or more observations that were off the trend. How does this point or points show up on each of the plots you drew in the previous part?

6. Any data points that concerned you earlier were actually errors. Create and save a new data frame that does not contain any of those data points.

7. Run a regression predicting height from foot length for your data set without errors. Obtain a plot of the residuals against fitted values and a normal quantile plot of the residuals for this regression.

8. Do you see any problems on the plots you drew in the previous part? Explain briefly.

9. Find a way to plot the data and both regression lines on the same plot, in such a way that you can see which regression line is which. If you get help from anything outside the course materials, cite your source(s).

10. Discuss briefly how removing the observation(s) that were errors has changed where the regression line goes, and whether that is what you expected.

My solutions follow:

## 18.15 Rainfall in California

The data in link are rainfall and other measurements for 30 weather stations in California. Our aim is to understand how well the annual rainfall at these stations (measured in inches) can be predicted from the other measurements, which are the altitude (in feet above sea level), the latitude (degrees north of the equator) and the distance from the coast (in miles).

1. Read the data into R. You’ll have to be careful here, since the values are space-delimited, but sometimes by more than one space, to make the columns line up. read_table, with filename or url, will read it in. One of the variables is called rainfall, so as long as you do not call the data frame that, you should be safe.

Solution

I used rains as the name of my data frame:

my_url="http://ritsokiguess.site/datafiles/calirain.txt"
rains=read_table(my_url)

── Column specification ────────────────────────────────────────────────────────
cols(
station = col_character(),
rainfall = col_double(),
altitude = col_double(),
latitude = col_double(),
fromcoast = col_double()
)

I have the right number of rows and columns.

I don’t need you to investigate the data yet (that happens in the next part), but this is interesting (to me):

rains

Some of the station names are two words, but they have been smooshed into one word, so that read_table will recognize them as a single thing. Someone had already done that for us, so I didn’t even have to do it myself.

If the station names had been two genuine words, a .csv would probably have been the best choice (the actual data values being separated by commas then, and not spaces).

$$\blacksquare$$

1. Make a boxplot of the rainfall figures, and explain why the values are reasonable. (A rainfall cannot be negative, and it is unusual for a annual rainfall to exceed 60 inches.) A ggplot boxplot needs something on the $$x$$-axis: the number 1 will do.

Solution

ggplot(rains,aes(y=rainfall,x=1))+geom_boxplot()

There is only one rainfall over 60 inches, and the smallest one is close to zero but positive, so that is good.

Another possible plot here is a histogram, since there is only one quantitative variable:

ggplot(rains, aes(x=rainfall))+geom_histogram(bins=7)

This clearly shows the rainfall value above 60 inches, but some other things are less clear: are those two rainfall values around 50 inches above or below 50, and are those six rainfall values near zero actually above zero? Extra: What stations have those extreme values? Should you wish to find out:

rains %>% filter(rainfall>60)

This is a place right on the Pacific coast, almost up into Oregon (it’s almost the northernmost of all the stations). So it makes sense that it would have a high rainfall, if anywhere does. (If you know anything about rainy places, you’ll probably think of Vancouver and Seattle, in the Pacific Northwest.) Here it is: link. Which station has less than 2 inches of annual rainfall?

rains %>% filter(rainfall<2)  

The name of the station is a clue: this one is in the desert. So you’d expect very little rain. Its altitude is negative, so it’s actually below sea level. This is correct. Here is where it is: link.

$$\blacksquare$$

1. Plot rainfall against each of the other quantitative variables (that is, not station).

Solution

That is, altitude, latitude and fromcoast. The obvious way to do this (perfectly acceptable) is one plot at a time:

ggplot(rains,aes(y=rainfall,x=altitude))+geom_point()
ggplot(rains,aes(y=rainfall,x=latitude))+geom_point()

and finally

ggplot(rains,aes(y=rainfall,x=fromcoast))+geom_point()

You can add a smooth trend to these if you want. Up to you. Just the points is fine with me.

Here is a funky way to get all three plots in one shot:

rains %>%
pivot_longer(altitude:fromcoast, names_to="xname",values_to="x") %>%
ggplot(aes(x=x,y=rainfall))+geom_point()+
facet_wrap(~xname,scales="free")

This always seems extraordinarily strange if you haven’t run into it before. The strategy is to put all the $$x$$-variables you want to plot into one column and then plot your $$y$$ against the $$x$$-column. Thus: make a column of all the $$x$$’s glued together, labelled by which $$x$$ they are, then plot $$y$$ against $$x$$ but make a different sub-plot or “facet” for each different $$x$$-name. The last thing is that each $$x$$ is measured on a different scale, and unless we take steps, all the sub-plots will have the same scale on each axis, which we don’t want.

I’m not sure I like how it came out, with three very tall plots. facet_wrap can also take an nrow or an ncol, which tells it how many rows or columns to use for the display. Here, for example, two columns because I thought three was too many:

rains %>%
pivot_longer(altitude:fromcoast, names_to="xname",values_to="x") %>%
ggplot(aes(x=x,y=rainfall))+geom_point()+
facet_wrap(~xname,scales="free",ncol=2)

Now, the three plots have come out about square, or at least “landscape”, which I like a lot better.

$$\blacksquare$$

1. Look at the relationship of each other variable with rainfall. Justify the assertion that latitude seems most strongly related with rainfall. Is that relationship positive or negative? linear? Explain briefly.

Solution

Let’s look at the three variables in turn:

• altitude: not much of anything. The stations near sea level have rainfall all over the place, though the three highest-altitude stations have the three highest rainfalls apart from Crescent City.

• latitude: there is a definite upward trend here, in that stations further north (higher latitude) are likely to have a higher rainfall. I’d call this trend linear (or, not obviously curved), though the two most northerly stations have one higher and one much lower rainfall than you’d expect.

• fromcoast: this is a weak downward trend, though the trend is spoiled by those three stations about 150 miles from the coast that have more than 40 inches of rainfall.

Out of those, only latitude seems to have any meaningful relationship with rainfall.

$$\blacksquare$$

1. Fit a regression with rainfall as the response variable, and latitude as your explanatory variable. What are the intercept, slope and R-squared values? Is there a significant relationship between rainfall and your explanatory variable? What does that mean?

Solution

Save your lm into a variable, since it will get used again later:

rainfall.1=lm(rainfall~latitude,data=rains)
summary(rainfall.1)

Call:
lm(formula = rainfall ~ latitude, data = rains)

Residuals:
Min      1Q  Median      3Q     Max
-27.297  -7.956  -2.103   6.082  38.262

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -113.3028    35.7210  -3.172  0.00366 **
latitude       3.5950     0.9623   3.736  0.00085 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 13.82 on 28 degrees of freedom
Multiple R-squared:  0.3326,    Adjusted R-squared:  0.3088
F-statistic: 13.96 on 1 and 28 DF,  p-value: 0.0008495

My intercept is $$-113.3$$, slope is $$3.6$$ and R-squared is $$0.33$$ or 33%. (I want you to pull these numbers out of the output and round them off to something sensible.) The slope is significantly nonzero, its P-value being 0.00085: rainfall really does depend on latitude, although not strongly so.

Extra: Of course, I can easily do the others as well, though you don’t have to:

rainfall.2=lm(rainfall~fromcoast,data=rains)
summary(rainfall.2)

Call:
lm(formula = rainfall ~ fromcoast, data = rains)

Residuals:
Min      1Q  Median      3Q     Max
-15.240  -9.431  -6.603   2.871  51.147

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) 23.77306    4.61296   5.154 1.82e-05 ***
fromcoast   -0.05039    0.04431  -1.137    0.265
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 16.54 on 28 degrees of freedom
Multiple R-squared:  0.04414,   Adjusted R-squared:   0.01
F-statistic: 1.293 on 1 and 28 DF,  p-value: 0.2651

Here, the intercept is 23.8, the slope is $$-0.05$$ and R-squared is a dismal 0.04 (4%). This is a way of seeing that this relationship is really weak, and it doesn’t even have a curve to the trend or anything that would compensate for this. I looked at the scatterplot again and saw that if it were not for the point bottom right which is furthest from the coast and has almost no rainfall, there would be almost no trend at all. The slope here is not significantly different from zero, with a P-value of 0.265.

Finally:

rainfall.3=lm(rainfall~altitude,data=rains)
summary(rainfall.3)

Call:
lm(formula = rainfall ~ altitude, data = rains)

Residuals:
Min      1Q  Median      3Q     Max
-20.620  -8.479  -2.729   4.555  58.271

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) 16.514799   3.539141   4.666  6.9e-05 ***
altitude     0.002394   0.001428   1.676    0.105
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 16.13 on 28 degrees of freedom
Multiple R-squared:  0.09121,   Adjusted R-squared:  0.05875
F-statistic:  2.81 on 1 and 28 DF,  p-value: 0.1048

The intercept is 16.5, the slope is 0.002 and the R-squared is 0.09 or 9%, also terrible. The P-value is 0.105, which is not small enough to be significant.

So it looks as if it’s only latitude that has any impact at all. This is the only explanatory variable with a significantly nonzero slope. On its own, at least.

$$\blacksquare$$

1. Fit a multiple regression predicting rainfall from all three of the other (quantitative) variables. Display the results. Comment is coming up later.

Solution

This, then:

rainfall.4=lm(rainfall~latitude+altitude+fromcoast,data=rains)
summary(rainfall.4)

Call:
lm(formula = rainfall ~ latitude + altitude + fromcoast, data = rains)

Residuals:
Min      1Q  Median      3Q     Max
-28.722  -5.603  -0.531   3.510  33.317

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -1.024e+02  2.921e+01  -3.505 0.001676 **
latitude     3.451e+00  7.949e-01   4.342 0.000191 ***
altitude     4.091e-03  1.218e-03   3.358 0.002431 **
fromcoast   -1.429e-01  3.634e-02  -3.931 0.000559 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 11.1 on 26 degrees of freedom
Multiple R-squared:  0.6003,    Adjusted R-squared:  0.5542
F-statistic: 13.02 on 3 and 26 DF,  p-value: 2.205e-05

$$\blacksquare$$

1. What is the R-squared for the regression of the last part? How does that compare with the R-squared of your regression in part (e)?

Solution

The R-squared is 0.60 (60%), which is quite a bit bigger than the R-squared of 0.33 (33%) we got back in (e).

$$\blacksquare$$

1. What do you conclude about the importance of the variables that you did not include in your model in (e)? Explain briefly.

Solution

Both variables altitude and fromcoast are significant in this regression, so they have something to add over and above latitude when it comes to predicting rainfall, even though (and this seems odd) they have no apparent relationship with rainfall on their own. Another way to say this is that the three variables work together as a team to predict rainfall, and together they do much better than any one of them can do by themselves.
This also goes to show that the scatterplots we began with don’t get to the heart of multi-variable relationships, because they are only looking at the variables two at a time.

$$\blacksquare$$

1. Make a suitable hypothesis test that the variables altitude and fromcoast significantly improve the prediction of rainfall over the use of latitude alone. What do you conclude?

Solution

This calls for anova. Feed this two fitted models, smaller (fewer explanatory variables) first. The null hypothesis is that the two models are equally good (so we should go with the smaller); the alternative is that the larger model is better, so that the extra complication is worth it:

anova(rainfall.1,rainfall.4)  

The P-value is small, so we reject the null in favour of the alternative: the regression with all three explanatory variables fits better than the one with just latitude, so the bigger model is the one we should go with.

If you have studied these things: this one is a “multiple-partial $$F$$-test”, for testing the combined significance of more than one $$x$$ but less than all the $$x$$’s.4

$$\blacksquare$$

## 18.16 Carbon monoxide in cigarettes

The (US) Federal Trade Commission assesses cigarettes according to their tar, nicotine and carbon monoxide contents. In a particular year, 25 brands were assessed. For each brand, the tar, nicotine and carbon monoxide (all in milligrams) were measured, along with the weight in grams. Our aim is to predict carbon monoxide from any or all of the other variables. The data are in link. These are aligned by column (except for the variable names), with more than one space between each column of data.

1. Read the data into R, and check that you have 25 observations and 4 variables.

Solution

This specification calls for read_table:

my_url <- "http://ritsokiguess.site/datafiles/ftccigar.txt"
cigs <- read_table(my_url)

── Column specification ────────────────────────────────────────────────────────
cols(
tar = col_double(),
nicotine = col_double(),
weight = col_double(),
co = col_double()
)
cigs

Yes, I have 25 observations on 4 variables indeed.

read_delim won’t work (try it and see what happens), because that would require the values to be separated by exactly one space.

$$\blacksquare$$

1. Run a regression to predict carbon monoxide from the other variables, and obtain a summary of the output.

Solution

The word “summary” is meant to be a big clue that summary is what you need:

cigs.1 <- lm(co ~ tar + nicotine + weight, data = cigs)
summary(cigs.1)

Call:
lm(formula = co ~ tar + nicotine + weight, data = cigs)

Residuals:
Min       1Q   Median       3Q      Max
-2.89261 -0.78269  0.00428  0.92891  2.45082

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)   3.2022     3.4618   0.925 0.365464
tar           0.9626     0.2422   3.974 0.000692 ***
nicotine     -2.6317     3.9006  -0.675 0.507234
weight       -0.1305     3.8853  -0.034 0.973527
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 1.446 on 21 degrees of freedom
Multiple R-squared:  0.9186,    Adjusted R-squared:  0.907
F-statistic: 78.98 on 3 and 21 DF,  p-value: 1.329e-11

$$\blacksquare$$

1. Which one of your explanatory variables would you remove from this regression? Explain (very) briefly. Go ahead and fit the regression without it, and describe how the change in R-squared from the regression in (b) was entirely predictable.

Solution

First, the $$x$$-variable to remove. The obvious candidate is weight, since it has easily the highest, and clearly non-significant, P-value. So, out it comes:

cigs.2 <- lm(co ~ tar + nicotine, data = cigs)
summary(cigs.2)

Call:
lm(formula = co ~ tar + nicotine, data = cigs)

Residuals:
Min       1Q   Median       3Q      Max
-2.89941 -0.78470 -0.00144  0.91585  2.43064

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)   3.0896     0.8438   3.662 0.001371 **
tar           0.9625     0.2367   4.067 0.000512 ***
nicotine     -2.6463     3.7872  -0.699 0.492035
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 1.413 on 22 degrees of freedom
Multiple R-squared:  0.9186,    Adjusted R-squared:  0.9112
F-statistic: 124.1 on 2 and 22 DF,  p-value: 1.042e-12

R-squared has dropped from 0.9186 to 0.9186! That is, taking out weight has not just had a minimal effect on R-squared; it’s not changed R-squared at all. This is because weight was so far from being significant: it literally had nothing to add.

Another way of achieving the same thing is via the function update, which takes a fitted model object and describes the change that you want to make:

cigs.2a <- update(cigs.1, . ~ . - weight)
summary(cigs.2a)

Call:
lm(formula = co ~ tar + nicotine, data = cigs)

Residuals:
Min       1Q   Median       3Q      Max
-2.89941 -0.78470 -0.00144  0.91585  2.43064

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)   3.0896     0.8438   3.662 0.001371 **
tar           0.9625     0.2367   4.067 0.000512 ***
nicotine     -2.6463     3.7872  -0.699 0.492035
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 1.413 on 22 degrees of freedom
Multiple R-squared:  0.9186,    Adjusted R-squared:  0.9112
F-statistic: 124.1 on 2 and 22 DF,  p-value: 1.042e-12

This can be shorter than describing the whole model again, as you do with the cigs.2 version of lm. The syntax is that you first specify a “base” fitted model object that you’re going to update. Because the model cigs.1 contains all the information about the kind of model it is, and which data frame the data come from, R already knows that this is a linear multiple regression and which $$x$$’s it contains. The second thing to describe is the change from the “base”. In this case, we want to use the same response variable and all the same explanatory variables that we had before, except for weight. This is specified by a special kind of model formula where . means “whatever was there before”: in English, “same response and same explanatories except take out weight”.

$$\blacksquare$$

1. Fit a regression predicting carbon monoxide from nicotine only, and display the summary.

Solution

As you would guess:

cigs.3 <- lm(co ~ nicotine, data = cigs)
summary(cigs.3)

Call:
lm(formula = co ~ nicotine, data = cigs)

Residuals:
Min      1Q  Median      3Q     Max
-3.3273 -1.2228  0.2304  1.2700  3.9357

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)   1.6647     0.9936   1.675    0.107
nicotine     12.3954     1.0542  11.759 3.31e-11 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 1.828 on 23 degrees of freedom
Multiple R-squared:  0.8574,    Adjusted R-squared:  0.8512
F-statistic: 138.3 on 1 and 23 DF,  p-value: 3.312e-11

$$\blacksquare$$

1. nicotine was far from being significant in the model of (c), and yet in the model of (d), it was strongly significant, and the R-squared value of (d) was almost as high as that of (c). What does this say about the importance of nicotine as an explanatory variable? Explain, as briefly as you can manage.

Solution

What this says is that you cannot say anything about the “importance” of nicotine without also describing the context that you’re talking about. By itself, nicotine is important, but when you have tar in the model, nicotine is not important: precisely, it now has nothing to add over and above the predictive value that tar has. You might guess that this is because tar and nicotine are “saying the same thing” in some fashion. We’ll explore that in a moment.

$$\blacksquare$$

1. Make a “pairs plot”: that is, scatter plots between all pairs of variables. This can be done by feeding the whole data frame into plot.5 Do you see any strong relationships that do not include co? Does that shed any light on the last part? Explain briefly (or “at length” if that’s how it comes out).

Solution

Plot the entire data frame:

plot(cigs)

We’re supposed to ignore co, but I comment that strong relationships between co and both of tar and nicotine show up here, along with weight being at most weakly related to anything else.

That leaves the relationship of tar and nicotine with each other. That also looks like a strong linear trend. When you have correlations between explanatory variables, it is called “multicollinearity”.

Having correlated $$x$$’s is trouble. Here is where we find out why. The problem is that when co is large, nicotine is large, and a large value of tar will come along with it. So we don’t know whether a large value of co is caused by a large value of tar or a large value of nicotine: there is no way to separate out their effects because in effect they are “glued together”.

You might know of this effect (in an experimental design context) as “confounding”: the effect of tar on co is confounded with the effect of nicotine on co, and you can’t tell which one deserves the credit for predicting co.

If you were able to design an experiment here, you could (in principle) manufacture a bunch of cigarettes with high tar; some of them would have high nicotine and some would have low. Likewise for low tar. Then the correlation between nicotine and tar would go away, their effects on co would no longer be confounded, and you could see unambiguously which one of the variables deserves credit for predicting co. Or maybe it depends on both, genuinely, but at least then you’d know.

We, however, have an observational study, so we have to make do with the data we have. Confounding is one of the risks we take when we work with observational data.

This was a “base graphics” plot. There is a way of doing a ggplot-style “pairs plot”, as this is called, thus:

library(GGally)
Registered S3 method overwritten by 'GGally':
method from
+.gg   ggplot2
cigs %>% ggpairs(progress = FALSE)

As ever, install.packages first, in the likely event that you don’t have this package installed yet. Once you do, though, I think this is a nicer way to get a pairs plot.

This plot is a bit more sophisticated: instead of just having the scatterplots of the pairs of variables in the row and column, it uses the diagonal to show a “kernel density” (a smoothed-out histogram), and upper-right it shows the correlation between each pair of variables. The three correlations between co, tar and nicotine are clearly the highest.

If you want only some of the columns to appear in your pairs plot, select them first, and then pass that data frame into ggpairs. Here, we found that weight was not correlated with anything much, so we can take it out and then make a pairs plot of the other variables:

cigs %>% select(-weight) %>% ggpairs(progress = FALSE)

The three correlations that remain are all very high, which is entirely consistent with the strong linear relationships that you see bottom left.

$$\blacksquare$$

## 18.17 Maximal oxygen uptake in young boys

A physiologist wanted to understand the relationship between physical characteristics of pre-adolescent boys and their maximal oxygen uptake (millilitres of oxygen per kilogram of body weight). The data are in link for a random sample of 10 pre-adolescent boys. The variables are (with units):

• uptake: Oxygen uptake (millitres of oxygen per kilogram of body weight)

• age: boy’s age (years)

• height: boy’s height (cm)

• weight: boy’s weight (kg)

• chest: chest depth (cm).

1. Read the data into R and confirm that you do indeed have 10 observations.

Solution

my_url <- "http://ritsokiguess.site/datafiles/youngboys.txt"
boys <- read_delim(my_url, " ")
Rows: 10 Columns: 5
── Column specification ────────────────────────────────────────────────────────
Delimiter: " "
dbl (5): uptake, age, height, weight, chest

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
boys

10 boys (rows) indeed.

$$\blacksquare$$

1. Fit a regression predicting oxygen uptake from all the other variables, and display the results.

Solution

Fitting four explanatory variables with only ten observations is likely to be pretty shaky, but we press ahead regardless:

boys.1 <- lm(uptake ~ age + height + weight + chest, data = boys)
summary(boys.1)

Call:
lm(formula = uptake ~ age + height + weight + chest, data = boys)

Residuals:
1         2         3         4         5         6         7         8
-0.020697  0.019741 -0.003649  0.038470 -0.023639 -0.026026  0.050459 -0.014380
9        10
0.004294 -0.024573

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -4.774739   0.862818  -5.534 0.002643 **
age         -0.035214   0.015386  -2.289 0.070769 .
height       0.051637   0.006215   8.308 0.000413 ***
weight      -0.023417   0.013428  -1.744 0.141640
chest        0.034489   0.085239   0.405 0.702490
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 0.03721 on 5 degrees of freedom
Multiple R-squared:  0.9675,    Adjusted R-squared:  0.9415
F-statistic:  37.2 on 4 and 5 DF,  p-value: 0.0006513

$$\blacksquare$$

1. (A one-mark question.) Would you say, on the evidence so far, that the regression fits well or badly? Explain (very) briefly.

Solution

R-squared of 0.97 (97%) is very high, so I’d say this regression fits very well. That’s all. I said “on the evidence so far” to dissuade you from overthinking this, or thinking that you needed to produce some more evidence. That, plus the fact that this was only one mark.

$$\blacksquare$$

1. It seems reasonable that an older boy should have a greater oxygen uptake, all else being equal. Is this supported by your output? Explain briefly.

Solution

If an older boy has greater oxygen uptake (the “all else equal” was a hint), the slope of age should be positive. It is not: it is $$-0.035$$, so it is suggesting (all else equal) that a greater age goes with a smaller oxygen uptake. The reason why this happens (which you didn’t need, but you can include it if you like) is that age has a non-small P-value of 0.07, so that the age slope is not significantly different from zero. With all the other variables, age has nothing to add over and above them, and we could therefore remove it.

$$\blacksquare$$

1. It seems reasonable that a boy with larger weight should have larger lungs and thus a statistically significantly larger oxygen uptake. Is that what happens here? Explain briefly.

Solution

Look at the P-value for weight. This is 0.14, not small, and so a boy with larger weight does not have a significantly larger oxygen uptake, all else equal. (The slope for weight is not significantly different from zero either.) I emphasized “statistically significant” to remind you that this means to do a test and get a P-value.

$$\blacksquare$$

1. Fit a model that contains only the significant explanatory variables from your first regression. How do the R-squared values from the two regressions compare? (The last sentence asks for more or less the same thing as the next part. Answer it either here or there. Either place is good.)

Solution

Only height is significant, so that’s the only explanatory variable we need to keep. I would just do the regression straight rather than using update here:

boys.2 <- lm(uptake ~ height, data = boys)
summary(boys.2)

Call:
lm(formula = uptake ~ height, data = boys)

Residuals:
Min        1Q    Median        3Q       Max
-0.069879 -0.033144  0.001407  0.009581  0.084012

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -3.843326   0.609198  -6.309 0.000231 ***
height       0.040718   0.004648   8.761 2.26e-05 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 0.05013 on 8 degrees of freedom
Multiple R-squared:  0.9056,    Adjusted R-squared:  0.8938
F-statistic: 76.75 on 1 and 8 DF,  p-value: 2.258e-05

If you want, you can use update here, which looks like this:

boys.2a <- update(boys.1, . ~ . - age - weight - chest)
summary(boys.2a)

Call:
lm(formula = uptake ~ height, data = boys)

Residuals:
Min        1Q    Median        3Q       Max
-0.069879 -0.033144  0.001407  0.009581  0.084012

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -3.843326   0.609198  -6.309 0.000231 ***
height       0.040718   0.004648   8.761 2.26e-05 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 0.05013 on 8 degrees of freedom
Multiple R-squared:  0.9056,    Adjusted R-squared:  0.8938
F-statistic: 76.75 on 1 and 8 DF,  p-value: 2.258e-05

This doesn’t go quite so smoothly here because there are three variables being removed, and it’s a bit of work to type them all.

$$\blacksquare$$

1. How has R-squared changed between your two regressions? Describe what you see in a few words.

Solution

R-squared has dropped by a bit, from 97% to 91%. (Make your own call: pull out the two R-squared numbers, and say a word or two about how they compare. I don’t much mind what you say: “R-squared has decreased (noticeably)”, “R-squared has hardly changed”. But say something.)

$$\blacksquare$$

1. Carry out a test comparing the fit of your two regression models. What do you conclude, and therefore what recommendation would you make about the regression that would be preferred?

Solution

The word “test” again implies something that produces a P-value with a null hypothesis that you might reject. In this case, the test that compares two models differing by more than one $$x$$ uses anova, testing the null hypothesis that the two regressions are equally good, against the alternative that the bigger (first) one is better. Feed anova two fitted model objects, smaller first:

anova(boys.2, boys.1)

This P-value of 0.123 is not small, so we do not reject the null hypothesis. There is not a significant difference in fit between the two models. Therefore, we should go with the smaller model boys.2 because it is simpler.

That drop in R-squared from 97% to 91% was, it turns out, not significant: the three extra variables could have produced a change in R-squared like that, even if they were worthless.6

If you have learned about “adjusted R-squared”, you might recall that this is supposed to go down only if the variables you took out should not have been taken out. But adjusted R-squared goes down here as well, from 94% to 89% (not quite as much, therefore). What happens is that adjusted R-squared is rather more relaxed about keeping variables than the anova $$F$$-test is; if we had used an $$\alpha$$ of something like 0.10, the decision between the two models would have been a lot closer, and this is reflected in the adjusted R-squared values.

$$\blacksquare$$

1. Obtain a table of correlations between all the variables in the data frame. Do this by feeding the whole data frame into cor. We found that a regression predicting oxygen uptake from just height was acceptably good. What does your table of correlations say about why that is? (Hint: look for all the correlations that are large.)

Solution

Correlations first:

cor(boys)
          uptake       age    height    weight     chest
uptake 1.0000000 0.1361907 0.9516347 0.6576883 0.7182659
age    0.1361907 1.0000000 0.3274830 0.2307403 0.1657523
height 0.9516347 0.3274830 1.0000000 0.7898252 0.7909452
weight 0.6576883 0.2307403 0.7898252 1.0000000 0.8809605
chest  0.7182659 0.1657523 0.7909452 0.8809605 1.0000000

The correlations with age are all on the low side, but all the other correlations are high, not just between uptake and the other variables, but between the explanatory variables as well.

Why is this helpful in understanding what’s going on? Well, imagine a boy with large height (a tall one). The regression boys.2 says that this alone is enough to predict that such a boy’s oxygen uptake is likely to be large, since the slope is positive. But the correlations tell you more: a boy with large height is also (somewhat) likely to be older (have large age), heavier (large weight) and to have larger chest cavity. So oxygen uptake does depend on those other variables as well, but once you know height you can make a good guess at their values; you don’t need to know them.

Further remarks: age has a low correlation with uptake, so its non-significance earlier appears to be “real”: it really does have nothing extra to say, because the other variables have a stronger link with uptake than age. Height, however, seems to be the best way of relating oxygen uptake to any of the other variables. I think the suppositions from earlier about relating oxygen uptake to “bigness”7 in some sense are actually sound, but age and weight and chest capture “bigness” worse than height does. Later, when you learn about Principal Components, you will see that the first principal component, the one that best captures how the variables vary together, is often “bigness” in some sense.

Another way to think about these things is via pairwise scatterplots. The nicest way to produce these is via ggpairs from package GGally:

boys %>% ggpairs(progress = FALSE)

A final remark: with five variables, we really ought to have more than ten observations (something like 50 would be better). But with more observations and the same correlation structure, the same issues would come up again, so the question would not be materially changed.

$$\blacksquare$$

## 18.18 Facebook friends and grey matter

Is there a relationship between the number of Facebook friends a person has, and the density of grey matter in the areas of the brain associated with social perception and associative memory? To find out, a 2012 study measured both of these variables for a sample of 40 students at City University in London (England). The data are at link. The grey matter density is on a $$z$$-score standardized scale. The values are separated by tabs.

The aim of this question is to produce a Quarto report that contains your answers to the questions below.

You should aim to make your report flow smoothly, so that it would be pleasant for a grader to read, and can stand on its own as an analysis (rather than just being the answer to a question that I set you). Some suggestions: give your report a title and arrange it into sections with an Introduction; add a small amount of additional text here and there explaining what you are doing and why. I don’t expect you to spend a large amount of time on this, but I do hope you will make some effort. (My report came out to 4 Word pages; HTML of course doesn’t have page numbers, but that’s a good kind of length to aim for.)

1. Read in the data and make a scatterplot for predicting the number of Facebook friends from the grey matter density. On your scatterplot, add a smooth trend.

Solution

Begin your document with a code chunk containing library(tidyverse). The data values are separated by tabs, which you will need to take into account:

my_url <- "http://ritsokiguess.site/datafiles/facebook.txt"
fb <- read_tsv(my_url)
Rows: 40 Columns: 2
── Column specification ────────────────────────────────────────────────────────
Delimiter: "\t"
dbl (2): GMdensity, FBfriends

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
fb
ggplot(fb, aes(x = GMdensity, y = FBfriends)) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

$$\blacksquare$$

1. Describe what you see on your scatterplot: is there a trend, and if so, what kind of trend is it? (Don’t get too taken in by the exact shape of your smooth trend.) Think “form, direction, strength”.

Solution

I’d say there seems to be a weak, upward, apparently linear trend. The points are not especially close to the trend, so I don’t think there’s any justification for calling this other than “weak”. (If you think the trend is, let’s say, “moderate”, you ought to say what makes you think that: for example, that the people with a lot of Facebook friends also tend to have a higher grey matter density. I can live with a reasonably-justified “moderate”.) The reason I said not to get taken in by the shape of the smooth trend is that this has a “wiggle” in it: it goes down again briefly in the middle. But this is likely a quirk of the data, and the trend, if there is any, seems to be an upward one.

$$\blacksquare$$

1. Fit a regression predicting the number of Facebook friends from the grey matter density, and display the output.

Solution

That looks like this. You can call the “fitted model object” whatever you like, but you’ll need to get the capitalization of the variable names correct:

fb.1 <- lm(FBfriends ~ GMdensity, data = fb)
summary(fb.1)

Call:
lm(formula = FBfriends ~ GMdensity, data = fb)

Residuals:
Min      1Q  Median      3Q     Max
-339.89 -110.01   -5.12   99.80  303.64

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)   366.64      26.35  13.916  < 2e-16 ***
GMdensity      82.45      27.58   2.989  0.00488 **
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 165.7 on 38 degrees of freedom
Multiple R-squared:  0.1904,    Adjusted R-squared:  0.1691
F-statistic: 8.936 on 1 and 38 DF,  p-value: 0.004882

I observe, though I didn’t ask you to, that the R-squared is pretty awful, going with a correlation of

sqrt(0.1904)
[1] 0.4363485

which would look like as weak of a trend as we saw.8

$$\blacksquare$$

1. Is the slope of your regression line significantly different from zero? What does that mean, in the context of the data?

Solution

The P-value of the slope is 0.005, which is less than 0.05. Therefore the slope is significantly different from zero. That means that the number of Facebook friends really does depend on the grey matter density, for the whole population of interest and not just the 40 students observed here (that were a sample from that population). I don’t mind so much what you think the population is, but it needs to be clear that the relationship applies to a population. Another way to approach this is to say that you would expect this relationship to show up again in another similar experiment. That also works, because it gets at the idea of reproducibility.

$$\blacksquare$$

1. Are you surprised by the results of parts (b) and (d)? Explain briefly.

Solution

I am surprised, because I thought the trend on the scatterplot was so weak that there would not be a significant slope. I guess there was enough of an upward trend to be significant, and with $$n=40$$ observations we were able to get a significant slope out of that scatterplot. With this many observations, even a weak correlation can be significantly nonzero. You can be surprised or not, but you need to have some kind of consideration of the strength of the trend on the scatterplot as against the significance of the slope. For example, if you decided that the trend was “moderate” in strength, you would be justified in being less surprised than I was. Here, there is the usual issue that we have proved that the slope is not zero (that the relationship is not flat), but we may not have a very clear idea of what the slope actually is. There are a couple of ways to get a confidence interval. The obvious one is to use R as a calculator and go up and down twice its standard error (to get a rough idea):

82.45 + 2 * 27.58 * c(-1, 1)
[1]  27.29 137.61

The c() thing is to get both confidence limits at once. The smoother way is this:

confint(fb.1)
                2.5 %   97.5 %
(Intercept) 313.30872 419.9810
GMdensity    26.61391 138.2836

Feed confint a “fitted model object” and it’ll give you confidence intervals (by default 95%) for all the parameters in it.

The confidence interval for the slope goes from about 27 to about 138. That is to say, a one-unit increase in grey matter density goes with an increase in Facebook friends of this much. This is not especially insightful: it’s bigger than zero (the test was significant), but other than that, it could be almost anything. This is where the weakness of the trend comes back to bite us. With this much scatter in our data, we need a much larger sample size to estimate accurately how big an effect grey matter density has.

$$\blacksquare$$

1. Obtain a scatterplot with the regression line on it.

Solution

Just a modification of (a):

ggplot(fb, aes(x = GMdensity, y = FBfriends)) + geom_point() +
geom_smooth(method = "lm")
geom_smooth() using formula = 'y ~ x'

$$\blacksquare$$

1. Obtain a plot of the residuals from the regression against the fitted values, and comment briefly on it.

Solution

This is, to my mind, the easiest way:

ggplot(fb.1, aes(x = .fitted, y = .resid)) + geom_point()

There is some “magic” here, since the fitted model object is not actually a data frame, but it works this way. That looks to me like a completely random scatter of points. Thus, I am completely happy with the straight-line regression that we fitted, and I see no need to improve it.

(You should make two points here: one, describe what you see, and two, what it implies about whether or not your regression is satisfactory.)

Compare that residual plot with this one:

ggplot(fb.1, aes(x = .fitted, y = .resid)) +
geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

Now, why did I try adding a smooth trend, and why is it not necessarily a good idea? The idea of a residual plot is that there should be no trend, and so the smooth trend curve ought to go straight across. The problem is that it will tend to wiggle, just by chance, as here: it looks as if it goes up and down before flattening out. But if you look at the points, they are all over the place, not close to the smooth trend at all. So the smooth trend is rather deceiving. Or, to put it another way, to indicate a real problem, the smooth trend would have to be a lot farther from flat than this one is. I’d call this one basically flat.

$$\blacksquare$$

## 18.19 Endogenous nitrogen excretion in carp

A paper in Fisheries Science reported on variables that affect “endogenous nitrogen excretion” or ENE in carp raised in Japan. A number of carp were divided into groups based on body weight, and each group was placed in a different tank. The mean body weight of the carp placed in each tank was recorded. The carp were then fed a protein-free diet three times daily for a period of 20 days. At the end of the experiment, the amount of ENE in each tank was measured, in milligrams of total fish body weight per day. (Thus it should not matter that some of the tanks had more fish than others, because the scaling is done properly.)

For this question, write a report in a Quarto document that answers the questions below and contains some narrative that describes your analysis. Create an HTML document from your Quarto document, as usual for an assignment.

Solution

Just this. Listing the data is up to you, but doing so and commenting that the values appear to be correct will improve your report.

my_url <- "http://ritsokiguess.site/datafiles/carp.txt"
carp <- read_delim(my_url, " ")
Rows: 10 Columns: 3
── Column specification ────────────────────────────────────────────────────────
Delimiter: " "
dbl (3): tank, bodyweight, ENE

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
carp

$$\blacksquare$$

1. Create a scatterplot of ENE (response) against bodyweight (explanatory). Add a smooth trend to your plot.

Solution

ggplot(carp, aes(x = bodyweight, y = ENE)) + geom_point() +
geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

This part is just about getting the plot. Comments are coming in a minute. Note that ENE is capital letters, so that ene will not work.

$$\blacksquare$$

1. Is there an upward or downward trend (or neither)? Is the relationship a line or a curve? Explain briefly.

Solution

The trend is downward: as bodyweight increases, ENE decreases. However, the decrease is rapid at first and then levels off, so the relationship is nonlinear. I want some kind of support for an assertion of non-linearity: anything that says that the slope or rate of decrease is not constant is good.

$$\blacksquare$$

1. Fit a straight line to the data, and obtain the R-squared for the regression.

Solution

lm. The first stage is to fit the straight line, saving the result in a variable, and the second stage is to look at the “fitted model object”, here via summary:

carp.1 <- lm(ENE ~ bodyweight, data = carp)
summary(carp.1)

Call:
lm(formula = ENE ~ bodyweight, data = carp)

Residuals:
Min     1Q Median     3Q    Max
-2.800 -1.957 -1.173  1.847  4.572

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) 11.40393    1.31464   8.675 2.43e-05 ***
bodyweight  -0.02710    0.01027  -2.640   0.0297 *
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 2.928 on 8 degrees of freedom
Multiple R-squared:  0.4656,    Adjusted R-squared:  0.3988
F-statistic: 6.971 on 1 and 8 DF,  p-value: 0.0297

Finally, you need to give me a (suitably rounded) value for R-squared: 46.6% or 47% or the equivalents as a decimal. I just need the value at this point. This kind of R-squared is actually pretty good for natural data, but the issue is whether we can improve it by fitting a non-linear model.9

$$\blacksquare$$

1. Obtain a residual plot (residuals against fitted values) for this regression. Do you see any problems? If so, what does that tell you about the relationship in the data?

Solution

This is the easiest way: feed the output of the regression straight into ggplot:

ggplot(carp.1, aes(x = .fitted, y = .resid)) + geom_point()

$$\blacksquare$$

1. Fit a parabola to the data (that is, including an $$x$$-squared term). Compare the R-squared values for the models in this part and part (d). Does that suggest that the parabola model is an improvement here over the linear model?

Solution

Add bodyweight-squared to the regression. Don’t forget the I():

carp.2 <- lm(ENE ~ bodyweight + I(bodyweight^2), data = carp)
summary(carp.2)

Call:
lm(formula = ENE ~ bodyweight + I(bodyweight^2), data = carp)

Residuals:
Min      1Q  Median      3Q     Max
-2.0834 -1.7388 -0.5464  1.3841  2.9976

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)     13.7127373  1.3062494  10.498 1.55e-05 ***
bodyweight      -0.1018390  0.0288109  -3.535  0.00954 **
I(bodyweight^2)  0.0002735  0.0001016   2.692  0.03101 *
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 2.194 on 7 degrees of freedom
Multiple R-squared:  0.7374,    Adjusted R-squared:  0.6624
F-statistic: 9.829 on 2 and 7 DF,  p-value: 0.009277

R-squared has gone up from 47% to 74%, a substantial improvement. This suggests to me that the parabola model is a substantial improvement.10

I try to avoid using the word “significant” in this context, since we haven’t actually done a test of significance.

The reason for the I() is that the up-arrow has a special meaning in lm, relating to interactions between factors (as in ANOVA), that we don’t want here. Putting I() around it means “use as is”, that is, raise bodyweight to power 2, rather than using the special meaning of the up-arrow in lm.

Because it’s the up-arrow that is the problem, this applies whenever you’re raising an explanatory variable to a power (or taking a reciprocal or a square root, say).

$$\blacksquare$$

1. Is the test for the slope coefficient for the squared term significant? What does this mean?

Solution

Look along the bodyweight-squared line to get a P-value of 0.031. This is less than the default 0.05, so it is significant. This means, in short, that the quadratic model is a significant improvement over the linear one.11 Said longer: the null hypothesis being tested is that the slope coefficient of the squared term is zero (that is, that the squared term has nothing to add over the linear model). This is rejected, so the squared term has something to add in terms of quality of prediction.

$$\blacksquare$$

1. Make the scatterplot of part (b), but add the fitted curve. Describe any way in which the curve fails to fit well.

Solution

This is a bit slippery, because the points to plot and the fitted curve are from different data frames. What you do in this case is to put a data= in one of the geoms, which says “don’t use the data frame that was in the ggplot, but use this one instead”. I would think about starting with the regression object carp.2 as my base data frame, since we want (or I want) to do two things with that: plot the fitted values and join them with lines. Then I want to add the original data, just the points:

ggplot(carp.2, aes(x = carp$bodyweight, y = .fitted), colour = "blue") + geom_line(colour = "blue") + geom_point(data = carp, aes(x = bodyweight, y = ENE)) This works, but is not very aesthetic, because the bodyweight that is plotted against the fitted values is in the wrong data frame, and so we have to use the dollar-sign thing to get it from the right one. A better way around this is “augment” the data with output from the regression object. This is done using augment from package broom: library(broom) carp.2a <- augment(carp.2, carp) carp.2a so now you see what carp.2a has in it, and then: g <- ggplot(carp.2a, aes(x = bodyweight, y = .fitted)) + geom_line(colour = "blue") + geom_point(aes(y = ENE)) This is easier coding: there are only two non-standard things. The first is that the fitted-value lines should be a distinct colour like blue so that you can tell them from the data points. The second thing is that for the second geom_point, the one that plots the data, the $$x$$ coordinate bodyweight is correct so that we don’t have to change that; we only have to change the $$y$$-coordinate, which is ENE. The plot is this: g Concerning interpretation, you have a number of possibilities here. The simplest is that the points in the middle are above the curve, and the points at the ends are below. (That is, negative residuals at the ends, and positive ones in the middle, which gives you a hint for the next part.) Another is that the parabola curve fails to capture the shape of the relationship; for example, I see nothing much in the data suggesting that the relationship should go back up, and even given that, the fitted curve doesn’t go especially near any of the points. I was thinking that the data should be fit better by something like the left half of an upward-opening parabola, but I guess the curvature on the left half of the plot suggests that it needs most of the left half of the parabola just to cover the left half of the plot. The moral of the story, as we see in the next part, is that the parabola is the wrong curve for the job. $$\blacksquare$$ 1. Obtain a residual plot for the parabola model. Do you see any problems with it? (If you do, I’m not asking you to do anything about them in this question, but I will.) $$\blacksquare$$ The same idea as before for the other residual plot. Use the fitted model object carp.2 as your data frame for the ggplot: ggplot(carp.2, aes(x = .fitted, y = .resid)) + geom_point() I think this is still a curve (or, it goes down and then sharply up at the end). Either way, there is still a pattern. That was all I needed, but as to what this means: our parabola was a curve all right, but it appears not to be the right kind of curve. I think the original data looks more like a hyperbola (a curve like $$y=1/x$$) than a parabola, in that it seems to decrease fast and then gradually to a limit, and that suggests, as in the class example, that we should try an asymptote model. Note how I specify it, with the I() thing again, since / has a special meaning to lm in the same way that ^ does: carp.3 <- lm(ENE ~ I(1 / bodyweight), data = carp) summary(carp.3)  Call: lm(formula = ENE ~ I(1/bodyweight), data = carp) Residuals: Min 1Q Median 3Q Max -1.29801 -0.12830 0.04029 0.26702 0.91707 Coefficients: Estimate Std. Error t value Pr(>|t|) (Intercept) 5.1804 0.2823 18.35 8.01e-08 *** I(1/bodyweight) 107.6690 5.8860 18.29 8.21e-08 *** --- Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1 Residual standard error: 0.6121 on 8 degrees of freedom Multiple R-squared: 0.9766, Adjusted R-squared: 0.9737 F-statistic: 334.6 on 1 and 8 DF, p-value: 8.205e-08 That fits extraordinarily well, with an R-squared up near 98%. The intercept is the asymptote, which suggests a (lower) limit of about 5.2 for ENE (in the limit for large bodyweight). We would have to ask the fisheries scientist whether this kind of thing is a reasonable biological mechanism. It says that a carp always has some ENE, no matter how big it gets, but a smaller carp will have a lot more. Does the fitted value plot look reasonable now? This is augment again since the fitted values and observed data come from different data frames: library(broom) augment(carp.3, carp) %>% ggplot(aes(x = bodyweight, y = .fitted)) + geom_line(colour = "blue") + geom_point(aes(y = ENE)) I’d say that does a really nice job of fitting the data. But it would be nice to have a few more tanks with large-bodyweight fish, to convince us that we have the shape of the trend right. And, as ever, the residual plot. That’s a lot easier than the plot we just did: ggplot(carp.3, aes(x = .fitted, y = .resid)) + geom_point() All in all, that looks pretty good (and certainly a vast improvement over the ones you got before). When you write up your report, you can make it flow better by writing it in a way that suggests that each thing was the obvious thing to do next: that is, that you would have thought to do it next, rather than me telling you what to do. My report (as a Quarto document) is at link. Download it, render it, play with it. $$\blacksquare$$ ## 18.20 Salaries of social workers Another salary-prediction question: does the number of years of work experience that a social worker has help to predict their salary? Data for 50 social workers are in link. 1. Read the data into R. Check that you have 50 observations on two variables. Also do something to check that the years of experience and annual salary figures look reasonable overall. Solution my_url <- "http://ritsokiguess.site/datafiles/socwork.txt" soc <- read_delim(my_url, " ") Rows: 50 Columns: 2 ── Column specification ──────────────────────────────────────────────────────── Delimiter: " " dbl (2): experience, salary ℹ Use spec() to retrieve the full column specification for this data. ℹ Specify the column types or set show_col_types = FALSE to quiet this message. soc That checks that we have the right number of observations; to check that we have sensible values, something like summary is called for: summary(soc)  experience salary Min. : 1.00 Min. :16105 1st Qu.:13.50 1st Qu.:36990 Median :20.00 Median :50948 Mean :18.12 Mean :50171 3rd Qu.:24.75 3rd Qu.:65204 Max. :28.00 Max. :99139  A person working in any field cannot have a negative number of years of experience, and cannot have more than about 40 years of experience (or else they would have retired). Our experience numbers fit that. Salaries had better be five or six figures, and salaries for social workers are not generally all that high, so these figures look reasonable. A rather more tidyverse way is this: soc %>% summarize(across(everything(), list(min = $$x) min(x), max = \(x) max(x)))) This gets the minimum and maximum of all the variables. I would have liked them arranged in a nice rectangle (min and max as rows, the variables as columns), but that’s not how this came out. We fix that shortly. The code so far uses across. This means to do something across multiple columns. In this case, we want to do the calculation on all the columns, so we use the select-helper everything. You can use any of the other select-helpers like starts_with, or you could do something like where(is.numeric) to do your summaries only on the quantitative columns (which would also work here). The thing after the everything() means “for each column selected, work out the min and max of it”; x is our name for “the variable we are looking at at the moment”. What, you want a nice rectangle? This is a pivot-longer, but a fancy version because the column names encode two kinds of things, a variable and a statistic. I took the view that I wanted variables in columns (as usual), and the different summary statistics in rows. This means that the first part of the column names we created above (eg. the salary part of salary_min) should stay in columns, and the rest of it should be pivoted longer. That means using the special name .value for the things that should stay as columns: soc %>% summarize(across(everything(), list(min = \(x) min(x), max = \(x) max(x)))) %>% pivot_longer(everything(), names_to = c(".value", "statistic"), names_sep = "_" ) Note that we’re using two simpler tools here, rather than one complicated one: first we get the summary statistics, and once we have that, we can do some tidying to get it arranged the way we want. Your first guess is likely to be to make it too long: soc %>% summarize(across(everything(), list(min = \(x) min(x), max = \(x) max(x)))) %>% pivot_longer(everything(), names_to = c("variable", "statistic"), names_sep = "_", values_to = "value" ) and then you’ll have to make it wider, or recall that you can do the thing with .value. We are working “columnwise”, doing something for each column, no matter how many there are. My go-to for this stuff is here. Another way to work is with the five-number summary. This gives a more nuanced picture of the data values we have.12 The base-R five-number summary looks like this: qq <- quantile(socexperience) qq  0% 25% 50% 75% 100% 1.00 13.50 20.00 24.75 28.00  This is what’s known as a “named vector”. The numbers on the bottom are the summaries themselves, and the names above say which percentile you are looking at. Unfortunately, the tidyverse doesn’t like names, so modelling after the above doesn’t quite work: soc %>% summarize(across(everything(), list(q = \(x) quantile(x)))) Warning: Returning more (or less) than 1 row per summarise() group was deprecated in dplyr 1.1.0. ℹ Please use reframe() instead. ℹ When switching from summarise() to reframe(), remember that reframe() always returns an ungrouped data frame and adjust accordingly. You can guess which percentile is which (they have to be in order), but this is not completely satisfactory. It also gives a warning because the summary is five numbers long, rather than only one (like the mean, for example), and this is not the preferred way to handle this. The warning mentions reframe, which is new (as in, less than a year old as I write this). Let’s see how it goes here: soc %>% reframe(q_exp = quantile(experience), q_sal = quantile(salary)) The idea is that reframe is like summarize, but it is designed for when your summary function returns more than one number, not just one number per group like mean or median do. This is not quite the best (I don’t see the percentiles and I have to repeat myself), but at least I no longer get a warning. Here’s how you do it with across: soc %>% reframe(across(everything(), \(x) enframe(quantile(x)), .unpack = TRUE)) The enframe turns a “named vector” (that is, a thing like my qq above) into a dataframe with two columns, one called name with the names (percentiles), and one called value with the values. By using across, you get those two columns for each variable, and you can see which of the five numbers is which percentile in each case. \(\blacksquare$$ 1. Make a scatterplot showing how salary depends on experience. Does the nature of the trend make sense? Solution The usual: ggplot(soc, aes(x = experience, y = salary)) + geom_point() As experience goes up, salary also goes up, as you would expect. Also, the trend seems more or less straight. $$\blacksquare$$ 1. Fit a regression predicting salary from experience, and display the results. Is the slope positive or negative? Does that make sense? Solution soc.1 <- lm(salary ~ experience, data = soc) summary(soc.1)  Call: lm(formula = salary ~ experience, data = soc) Residuals: Min 1Q Median 3Q Max -17666.3 -5498.2 -726.7 4667.7 27811.6 Coefficients: Estimate Std. Error t value Pr(>|t|) (Intercept) 11368.7 3160.3 3.597 0.000758 *** experience 2141.4 160.8 13.314 < 2e-16 *** --- Signif. codes: 0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1 Residual standard error: 8642 on 48 degrees of freedom Multiple R-squared: 0.7869, Adjusted R-squared: 0.7825 F-statistic: 177.3 on 1 and 48 DF, p-value: < 2.2e-16 The slope is (significantly) positive, which squares with our guess (more experience goes with greater salary), and also the upward trend on the scatterplot. The value of the slope is about 2,000; this means that one more year of experience goes with about a$2,000 increase in salary.

$$\blacksquare$$

1. Obtain and plot the residuals against the fitted values. What problem do you see?

Solution

The easiest way to do this with ggplot is to plot the regression object (even though it is not actually a data frame), and plot the .fitted and .resid columns in it, not forgetting the initial dots:

ggplot(soc.1, aes(x = .fitted, y = .resid)) + geom_point()

I see a “fanning-out”: the residuals are getting bigger in size (further away from zero) as the fitted values get bigger. That is, when the (estimated) salary gets larger, it also gets more variable.

Fanning-out is sometimes hard to see. What you can do if you suspect that it might have happened is to plot the absolute value of the residuals against the fitted values. The absolute value is the residual without its plus or minus sign, so if the residuals are getting bigger in size, their absolute values are getting bigger. That would look like this:

ggplot(soc.1, aes(x = .fitted, y = abs(.resid))) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

I added a smooth trend to this to help us judge whether the absolute-value-residuals are getting bigger as the fitted values get bigger. It looks to me as if the overall trend is an increasing one, apart from those few small fitted values that have larger-sized residuals. Don’t get thrown off by the kinks in the smooth trend. Here is a smoother version:

ggplot(soc.1, aes(x = .fitted, y = abs(.resid))) + geom_point() + geom_smooth(span = 2)
geom_smooth() using method = 'loess' and formula = 'y ~ x'

The larger fitted values, according to this, have residuals larger in size.

The thing that controls the smoothness of the smooth trend is the value of span in geom_smooth. The default is 0.75. The larger the value you use, the smoother the trend; the smaller, the more wiggly. I’m inclined to think that the default value is a bit too small. Possibly this value is too big, but it shows you the idea.

$$\blacksquare$$

1. The problem you unearthed in the previous part is often helped by a transformation. Run Box-Cox on your data to find a suitable transformation. What transformation is suggested?

Solution

You’ll need to load (and install if necessary) the package MASS that contains boxcox:

library(MASS)

When you run this, you may see a warning containing the word “masked”. I talk about that below.

boxcox(salary ~ experience, data = soc)

That one looks like $$\lambda=0$$ or log. You could probably also justify fourth root (power 0.25), but log is a very common transformation, which people won’t need much persuasion to accept.

Extra: There’s one annoyance with MASS: it has a select (which I have never used), and if you load tidyverse first and MASS second, as I have done here, when you mean to run the column-selection select, it will actually run the select that comes from MASS, and give you an error that you will have a terrible time debugging. That’s what that “masked” message was when you loaded MASS. This is a great place to learn about the conflicted package. See here for how it works. (Scroll down to under the list of files.)

If you want to insist on something like “the select that lives in dplyr”, you can do that by saying dplyr::select. But this is kind of cumbersome if you don’t need to do it.

$$\blacksquare$$

1. Use your transformed response in a regression, showing the summary.

Solution

You can do the transformation right in the lm, as I do below, or if you prefer, you can create a new column that is the log-salary and then use that in the lm. Either way is good:

soc.3 <- lm(log(salary) ~ experience, data = soc)
summary(soc.3)

Call:
lm(formula = log(salary) ~ experience, data = soc)

Residuals:
Min       1Q   Median       3Q      Max
-0.35435 -0.09046 -0.01725  0.09739  0.26355

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) 9.841315   0.056356  174.63   <2e-16 ***
experience  0.049979   0.002868   17.43   <2e-16 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 0.1541 on 48 degrees of freedom
Multiple R-squared:  0.8635,    Adjusted R-squared:  0.8607
F-statistic: 303.7 on 1 and 48 DF,  p-value: < 2.2e-16

$$\blacksquare$$

1. Obtain and plot the residuals against the fitted values for this regression. Do you seem to have solved the problem with the previous residual plot?

Solution

As we did before, treating the regression object as if it were a data frame:

ggplot(soc.3, aes(x = .fitted, y = .resid)) + geom_point()

That, to my mind, is a horizontal band of points, so I would say yes, I have solved the fanning out.

One concern I have about the residuals is that there seem to be a couple of very negative values: that is, are the residuals normally distributed as they should be? Well, that’s easy enough to check:

ggplot(soc.3, aes(sample = .resid)) + stat_qq() + stat_qq_line()

The issues here are that those bottom two values are a bit too low, and the top few values are a bit bunched up (that curve at the top). It is really not bad, though, so I am making the call that I don’t think I needed to worry. Note that the transformation we found here is the same as the log-salary used by the management consultants in the backward-elimination question, and with the same effect: an extra year of experience goes with a percent increase in salary.

What increase? Well, the slope is about 0.05, so adding a year of experience is predicted to increase log-salary by 0.05, or to multiply actual salary by

exp(0.05)
[1] 1.051271

or to increase salary by about 5%.13

$$\blacksquare$$

## 18.21 Predicting volume of wood in pine trees

In forestry, the financial value of a tree is the volume of wood that it contains. This is difficult to estimate while the tree is still standing, but the diameter is easy to measure with a tape measure (to measure the circumference) and a calculation involving $$\pi$$, assuming that the cross-section of the tree is at least approximately circular. The standard measurement is “diameter at breast height” (that is, at the height of a human breast or chest), defined as being 4.5 feet above the ground.

Several pine trees had their diameter measured shortly before being cut down, and for each tree, the volume of wood was recorded. The data are in link. The diameter is in inches and the volume is in cubic inches. Is it possible to predict the volume of wood from the diameter?

1. Read the data into R and display the values (there are not very many).

Solution

Observe that the data values are separated by spaces, and therefore that read_delim will do it:

my_url <- "http://ritsokiguess.site/datafiles/pinetrees.txt"
trees <- read_delim(my_url, " ")
Rows: 10 Columns: 2
── Column specification ────────────────────────────────────────────────────────
Delimiter: " "
dbl (2): diameter, volume

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
trees

That looks like the data file.

$$\blacksquare$$

1. Make a suitable plot.

Solution

No clues this time. You need to recognize that you have two quantitative variables, so that a scatterplot is called for. Also, the volume is the response, so that should go on the $$y$$-axis:

ggplot(trees, aes(x = diameter, y = volume)) + geom_point()

You can put a smooth trend on it if you like, which would look like this:

ggplot(trees, aes(x = diameter, y = volume)) +
geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

I’ll take either of those for this part, though I think the smooth trend actually obscures the issue here (because there is not so much data).

$$\blacksquare$$

1. Describe what you learn from your plot about the relationship between diameter and volume, if anything.

Solution

The word “relationship” offers a clue that a scatterplot would have been a good idea, if you hadn’t realized by now. I am guided by “form, direction, strength” in looking at a scatterplot:

• Form: it is an apparently linear relationship.

• Direction: it is an upward trend: that is, a tree with a larger diameter also has a larger volume of wood. (This is not very surprising.)

• Strength: I’d call this a strong (or moderate-to-strong) relationship. (We’ll see in a minute what the R-squared is.)

You don’t need to be as formal as this, but you do need to get at the idea that it is an upward trend, apparently linear, and at least fairly strong.14

$$\blacksquare$$

1. Fit a (linear) regression, predicting volume from diameter, and obtain the summary. How would you describe the R-squared?

Solution

My naming convention is (usually) to call the fitted model object by the name of the response variable and a number.15

volume.1 <- lm(volume ~ diameter, data = trees)
summary(volume.1)

Call:
lm(formula = volume ~ diameter, data = trees)

Residuals:
Min      1Q  Median      3Q     Max
-36.497  -9.982   1.751   8.959  28.139

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -191.749     23.954  -8.005 4.35e-05 ***
diameter      10.894      0.801  13.600 8.22e-07 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 20.38 on 8 degrees of freedom
Multiple R-squared:  0.9585,    Adjusted R-squared:  0.9534
F-statistic:   185 on 1 and 8 DF,  p-value: 8.217e-07

R-squared is nearly 96%, so the relationship is definitely a strong one.

I also wanted to mention the broom package, which was installed with the tidyverse but which you need to load separately. It provides two handy ways to summarize a fitted model (regression, analysis of variance or whatever):

library(broom)
glance(volume.1)

This gives a one-line summary of a model, including things like R-squared. This is handy if you’re fitting more than one model, because you can collect the one-line summaries together into a data frame and eyeball them.

The other summary is this one:

tidy(volume.1)

This gives a table of intercepts, slopes and their P-values, but the value to this one is that it is a data frame, so if you want to pull anything out of it, you know how to do that:16

tidy(volume.1) %>% filter(term == "diameter")

This gets the estimated slope and its P-value, without worrying about the corresponding things for the intercept, which are usually of less interest anyway.

$$\blacksquare$$

1. Draw a graph that will help you decide whether you trust the linearity of this regression. What do you conclude? Explain briefly.

Solution

The thing I’m fishing for is a residual plot (of the residuals against the fitted values), and on it you are looking for a random mess of nothingness:

ggplot(volume.1, aes(x = .fitted, y = .resid)) + geom_point()

Make a call. You could say that there’s no discernible pattern, especially with such a small data set, and therefore that the regression is fine. Or you could say that there is fanning-in: the two points on the right have residuals close to 0 while the points on the left have residuals larger in size. Say something.

I don’t think you can justify a curve or a trend, because the residuals on the left are both positive and negative.

My feeling is that the residuals on the right are close to 0 because these points have noticeably larger diameter than the others, and they are influential points in the regression that will pull the line closer to themselves. This is why their residuals are close to zero. But I am happy with either of the points made in the paragraph under the plot.

$$\blacksquare$$

1. What would you guess would be the volume of a tree of diameter zero? Is that what the regression predicts? Explain briefly.

Solution

Logically, a tree that has diameter zero is a non-existent tree, so its volume should be zero as well. In the regression, the quantity that says what volume is when diameter is zero is the intercept. Here the intercept is $$-192$$, which is definitely not zero. In fact, if you look at the P-value, the intercept is significantly less than zero. Thus, the model makes no logical sense for trees of small diameter. The smallest tree in the data set has diameter 18, which is not really small, I suppose, but it is a little disconcerting to have a model that makes no logical sense.

$$\blacksquare$$

1. A simple way of modelling a tree’s shape is to pretend it is a cone, like this, but probably taller and skinnier:

with its base on the ground. What is the relationship between the diameter (at the base) and volume of a cone? (If you don’t remember, look it up. You’ll probably get a formula in terms of the radius, which you’ll have to convert. Cite the website you used.)

Solution

According to link, the volume of a cone is $$V=\pi r^2h/3$$, where $$V$$ is the volume, $$r$$ is the radius (at the bottom of the cone) and $$h$$ is the height. The diameter is twice the radius, so replace $$r$$ by $$d/2$$, $$d$$ being the diameter. A little algebra gives $V = \pi d^2 h / 12.$

$$\blacksquare$$

1. Fit a regression model that predicts volume from diameter according to the formula you obtained in the previous part. You can assume that the trees in this data set are of similar heights, so that the height can be treated as a constant.
Display the results.

Solution

According to my formula, the volume depends on the diameter squared, which I include in the model thus:

volume.2 <- lm(volume ~ I(diameter^2), data = trees)
summary(volume.2)

Call:
lm(formula = volume ~ I(diameter^2), data = trees)

Residuals:
Min      1Q  Median      3Q     Max
-29.708  -9.065  -5.722   3.032  40.816

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)   -30.82634   13.82243   -2.23   0.0563 .
I(diameter^2)   0.17091    0.01342   12.74 1.36e-06 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 21.7 on 8 degrees of freedom
Multiple R-squared:  0.953, Adjusted R-squared:  0.9471
F-statistic: 162.2 on 1 and 8 DF,  p-value: 1.359e-06

This adds an intercept as well, which is fine (there are technical difficulties around removing the intercept).

That’s as far as I wanted you to go, but (of course) I have a few comments.

The intercept here is still negative, but not significantly different from zero, which is a step forward. The R-squared for this regression is very similar to that from our linear model (the one for which the intercept made no sense). So, from that point of view, either model predicts the data well. I should look at the residuals from this one:

ggplot(volume.2, aes(x = .fitted, y = .resid)) + geom_point()

I really don’t think there are any problems there.

Now, I said to assume that the trees are all of similar height. This seems entirely questionable, since the trees vary quite a bit in diameter, and you would guess that trees with bigger diameter would also be taller. It seems more plausible that the same kind of trees (pine trees in this case) would have the same “shape”, so that if you knew the diameter you could predict the height, with larger-diameter trees being taller. Except that we don’t have the heights here, so we can’t build a model for that.

So I went looking in the literature. I found this paper: link. This gives several models for relationships between volume, diameter and height. In the formulas below, there is an implied “plus error” on the right, and the $$\alpha_i$$ are parameters to be estimated.

For predicting height from diameter (equation 1 in paper):

$h = \exp(\alpha_1+\alpha_2 d^{\alpha_3})$

For predicting volume from height and diameter (equation 6):

$V = \alpha_1 d^{\alpha_2} h^{\alpha_3}$

This is a take-off on our assumption that the trees were cone-shaped, with cone-shaped trees having $$\alpha_1=\pi/12$$, $$\alpha_2=2$$ and $$\alpha_3=1$$. The paper uses different units, so $$\alpha_1$$ is not comparable, but $$\alpha_2$$ and $$\alpha_3$$ are (as estimated from the data in the paper, which were for longleaf pine) quite close to 2 and 1.

Last, the actual relationship that helps us: predicting volume from just diameter (equation 5):

$V = \alpha_1 d^{\alpha_2}$

This is a power law type of relationship. For example, if you were willing to pretend that a tree was a cone with height proportional to diameter (one way of getting at the idea of a bigger tree typically being taller, instead of assuming constant height as we did), that would imply $$\alpha_2=3$$ here.

This is non-linear as it stands, but we can bash it into shape by taking logs:

$\ln V = \ln \alpha_1 + \alpha_2 \ln d$

so that log-volume has a linear relationship with log-diameter and we can go ahead and estimate it:

volume.3 <- lm(log(volume) ~ log(diameter), data = trees)
summary(volume.3)

Call:
lm(formula = log(volume) ~ log(diameter), data = trees)

Residuals:
Min       1Q   Median       3Q      Max
-0.40989 -0.22341  0.01504  0.10459  0.53596

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)    -5.9243     1.1759  -5.038    0.001 **
log(diameter)   3.1284     0.3527   8.870 2.06e-05 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 0.3027 on 8 degrees of freedom
Multiple R-squared:  0.9077,    Adjusted R-squared:  0.8962
F-statistic: 78.68 on 1 and 8 DF,  p-value: 2.061e-05

The parameter that I called $$\alpha_2$$ above is the slope of this model, 3.13. This is a bit different from the figure in the paper, which was 2.19. I think these are comparable even though the other parameter is not (again, measurements in different units, plus, this time we need to take the log of it). I think the “slopes” are comparable because we haven’t estimated our slope all that accurately:

confint(volume.3)
                  2.5 %    97.5 %
(Intercept)   -8.635791 -3.212752
log(diameter)  2.315115  3.941665

From 2.3 to 3.9. It is definitely not zero, but we are rather less sure about what it is, and 2.19 is not completely implausible.

The R-squared here, though it is less than the other ones we got, is still high. The residuals are these:

ggplot(volume.3, aes(x = .fitted, y = .resid)) + geom_point()

which again seem to show no problems. The residuals are smaller in size now because of the log transformation: the actual and predicted log-volumes are smaller numbers than the actual and predicted volumes, so the residuals are now closer to zero.

Does this model behave itself at zero? Well, roughly at least: if the diameter is very small, its log is very negative, and the predicted log-volume is also very negative (the slope is positive). So the predicted actual volume will be close to zero. If you want to make that mathematically rigorous, you can take limits, but that’s the intuition. We can also do some predictions: set up a data frame that has a column called diameter with some diameters to predict for:

d <- tibble(diameter = c(1, 2, seq(5, 50, 5)))
d

and then feed that into predictionsfrom package marginaleffects:

p <- cbind(predictions(volume.3, newdata = d))
p %>% select(diameter, estimate, conf.low, conf.high) -> pp
pp

These are predicted log-volumes, so we’d better anti-log them. log in R is natural logs, so this is inverted using exp. The ends of the confidence intervals can be exp-ed as well, which I do all at once:

pp %>% mutate(across(-diameter, $$x) exp(x))) For a diameter near zero, the predicted volume appears to be near zero as well. If you don’t like the scientific notation: pp %>% mutate(across(-diameter, \(x) exp(x))) %>% mutate(across(-diameter, \(x) format(x, scientific = FALSE))) Note now that these, though they look like numbers, are actually text, so if you want to display numbers in non-scientific notation like this, do it at the very end, after you have finished any calculations with the numbers. I mentioned broom earlier. We can make a data frame out of the one-line summaries of our three models: bind_rows(glance(volume.1), glance(volume.2), glance(volume.3)) (I mistakenly put glimpse instead of glance there the first time. The former is for a quick look at a data frame, while the latter is for a quick look at a model.) The three R-squareds are all high, with the one from the third model being a bit lower as we saw before. My code is rather repetitious. There has to be a way to streamline it. I was determined to find out how. My solution involves putting the three models in a list-column, and then using rowwise to get the glance output for each one. tibble(i = 1:3, model = list(volume.1, volume.2, volume.3)) %>% rowwise() %>% mutate(glances = list(glance(model))) %>% unnest(glances) I almost got caught by forgetting the list on the definition of glances. I certainly need it, because the output from glance is a (one-row) dataframe, not a single number. It works. You see the three R-squared values in the first column of numbers. The third model is otherwise a lot different from the others because it has a different response variable. Other thoughts: How might you measure or estimate the height of a tree (other than by climbing it and dropping a tape measure down)? One way, that works if the tree is fairly isolated, is to walk away from its base. Periodically, you point at the top of the tree, and when the angle between your arm and the ground reaches 45 degrees, you stop walking. (If it’s greater than 45 degrees, you walk further away, and if it’s less, you walk back towards the tree.) The distance between you and the base of the tree is then equal to the height of the tree, and if you have a long enough tape measure you can measure it. The above works because the tangent of 45 degrees is 1. If you have a device that will measure the actual angle,17 you can be any distance away from the tree, point the device at the top, record the angle, and do some trigonometry to estimate the height of the tree (to which you add the height of your eyes). \(\blacksquare$$

## 18.22 Tortoise shells and eggs

A biologist measured the length of the carapace (shell) of female tortoises, and then x-rayed the tortoises to count how many eggs they were carrying. The length is measured in millimetres. The data are in link. The biologist is wondering what kind of relationship, if any, there is between the carapace length (as an explanatory variable) and the number of eggs (as a response variable).

1. Read in the data, and check that your values look reasonable.

Solution

Look at the data first. The columns are aligned and separated by more than one space, so it’s read_table:

my_url <- "http://ritsokiguess.site/datafiles/tortoise-eggs.txt"
tortoises <- read_table(my_url)

── Column specification ────────────────────────────────────────────────────────
cols(
length = col_double(),
eggs = col_double()
)
tortoises

Those look the same as the values in the data file. (Some comment is needed here. I don’t much mind what, but something that suggests that you have eyeballed the data and there are no obvious problems: that is what I am looking for.)

$$\blacksquare$$

1. Obtain a scatterplot, with a smooth trend, of the data.

Solution

Something like this:

ggplot(tortoises, aes(x = length, y = eggs)) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

$$\blacksquare$$

1. The biologist expected that a larger tortoise would be able to carry more eggs. Is that what the scatterplot is suggesting? Explain briefly why or why not.

Solution

The biologist’s expectation is of an upward trend. But it looks as if the trend on the scatterplot is up, then down, ie. a curve rather than a straight line. So this is not what the biologist was expecting.

$$\blacksquare$$

1. Fit a straight-line relationship and display the summary.

Solution

tortoises.1 <- lm(eggs ~ length, data = tortoises)
summary(tortoises.1)

Call:
lm(formula = eggs ~ length, data = tortoises)

Residuals:
Min      1Q  Median      3Q     Max
-6.7790 -1.1772 -0.0065  2.0487  4.8556

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -0.43532   17.34992  -0.025    0.980
length       0.02759    0.05631   0.490    0.631

Residual standard error: 3.411 on 16 degrees of freedom
Multiple R-squared:  0.01478,   Adjusted R-squared:  -0.0468
F-statistic:  0.24 on 1 and 16 DF,  p-value: 0.6308

I didn’t ask for a comment, but feel free to observe that this regression is truly awful, with an R-squared of less than 2% and a non-significant effect of length.

$$\blacksquare$$

1. Add a squared term to your regression, fit that and display the summary.

Solution

The I() is needed because the raise-to-a-power symbol has a special meaning in a model formula, and we want to not use that special meaning:

tortoises.2 <- lm(eggs ~ length + I(length^2), data = tortoises)
summary(tortoises.2)

Call:
lm(formula = eggs ~ length + I(length^2), data = tortoises)

Residuals:
Min      1Q  Median      3Q     Max
-4.0091 -1.8480 -0.1896  2.0989  4.3605

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -8.999e+02  2.703e+02  -3.329  0.00457 **
length       5.857e+00  1.750e+00   3.347  0.00441 **
I(length^2) -9.425e-03  2.829e-03  -3.332  0.00455 **
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 2.671 on 15 degrees of freedom
Multiple R-squared:  0.4338,    Adjusted R-squared:  0.3583
F-statistic: 5.747 on 2 and 15 DF,  p-value: 0.01403

Another way is to use update:

tortoises.2a <- update(tortoises.1, . ~ . + I(length^2))
summary(tortoises.2a)

Call:
lm(formula = eggs ~ length + I(length^2), data = tortoises)

Residuals:
Min      1Q  Median      3Q     Max
-4.0091 -1.8480 -0.1896  2.0989  4.3605

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -8.999e+02  2.703e+02  -3.329  0.00457 **
length       5.857e+00  1.750e+00   3.347  0.00441 **
I(length^2) -9.425e-03  2.829e-03  -3.332  0.00455 **
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 2.671 on 15 degrees of freedom
Multiple R-squared:  0.4338,    Adjusted R-squared:  0.3583
F-statistic: 5.747 on 2 and 15 DF,  p-value: 0.01403

$$\blacksquare$$

1. Is a curve better than a line for these data? Justify your answer in two ways: by comparing a measure of fit, and by doing a suitable test of significance.

Solution

An appropriate measure of fit is R-squared. For the straight line, this is about 0.01, and for the regression with the squared term it is about 0.43. This tells us that a straight line fits appallingly badly, and that a curve fits a lot better. This doesn’t do a test, though. For that, look at the slope of the length-squared term in the second regression; in particular, look at its P-value. This is 0.0045, which is small: the squared term is necessary, and taking it out would be a mistake. The relationship really is curved, and trying to describe it with a straight line would be a big mistake.

$$\blacksquare$$

1. Make a residual plot for the straight line model: that is, plot the residuals against the fitted values. Does this echo your conclusions of the previous part? In what way? Explain briefly.

Solution

Plot the things called .fitted and .resid from the regression object, which is not a data frame but you can treat it as if it is for this:

ggplot(tortoises.1, aes(x = .fitted, y = .resid)) + geom_point()

Up to you whether you put a smooth trend on it or not:

ggplot(tortoises.1, aes(x = .fitted, y = .resid)) + geom_point() +
geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

Looking at the plot, you see a curve, up and down. The most negative residuals go with small or large fitted values; when the fitted value is in the middle, the residual is usually positive. A curve on the residual plot indicates a curve in the actual relationship. We just found above that a curve does fit a lot better, so this is all consistent.

Aside: the grey “envelope” is wide, so there is a lot of scatter on the residual plot. The grey envelope almost contains zero all the way across, so the evidence for a curve (or any other kind of trend) is not all that strong, based on this plot. This is in great contrast to the regression with length-squared, where the length-squared term is definitely necessary.

That was all I wanted, but you can certainly look at other plots. Normal quantile plot of the residuals:

ggplot(tortoises.1, aes(sample = .resid)) + stat_qq() + stat_qq_line()

This is not the best: the low values are a bit too low, so that the whole picture is (a little) skewed to the left.18

Another plot you can make is to assess fan-out: you plot the absolute value19 of the residuals against the fitted values. The idea is that if there is fan-out, the absolute value of the residuals will get bigger:

ggplot(tortoises.1, aes(x = .fitted, y = abs(.resid))) + geom_point() +
geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

I put the smooth curve on as a kind of warning: it looks as if the size of the residuals goes down and then up again as the fitted values increase. But the width of the grey “envelope” and the general scatter of the points suggests that there is really not much happening here at all. On a plot of residuals, the grey envelope is really more informative than the blue smooth trend. On this one, there is no evidence of any fan-out (or fan-in).

$$\blacksquare$$

## 18.23 Roller coasters

A poll on the Discovery Channel asked people to nominate the best roller-coasters in the United States. We will examine the 10 roller-coasters that received the most votes. Two features of a roller-coaster that are of interest are the distance it drops from start to finish, measured here in feet20 and the duration of the ride, measured in seconds. Is it true that roller-coasters with a bigger drop also tend to have a longer ride? The data are at link.21

1. Read the data into R and verify that you have a sensible number of rows and columns.

Solution

A .csv, so the usual for that:

my_url <- "http://ritsokiguess.site/datafiles/coasters.csv"
coasters <- read_csv(my_url)
Rows: 10 Columns: 4
── Column specification ────────────────────────────────────────────────────────
Delimiter: ","
chr (2): coaster_name, state
dbl (2): drop, duration

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
coasters

The number of marks for this kind of thing has been decreasing through the course, since by now you ought to have figured out how to do it without looking it up.

There are 10 rows for the promised 10 roller-coasters, and there are several columns: the drop for each roller-coaster and the duration of its ride, as promised, as well as the name of each roller-coaster and the state that it is in. (A lot of them seem to be in Ohio, for some reason that I don’t know.) So this all looks good.

$$\blacksquare$$

1. Make a scatterplot of duration (response) against drop (explanatory), labelling each roller-coaster with its name in such a way that the labels do not overlap. Add a regression line to your plot.

Solution

The last part, about the labels not overlapping, is an invitation to use ggrepel, which is the way I’d recommend doing this. (If not, you have to do potentially lots of work organizing where the labels sit relative to the points, which is time you probably don’t want to spend.) Thus:

library(ggrepel)
ggplot(coasters, aes(x = drop, y = duration, label = coaster_name)) +
geom_point() + geom_text_repel() +
geom_smooth(method = "lm", se = FALSE)
geom_smooth() using formula = 'y ~ x'
Warning: The following aesthetics were dropped during statistical transformation: label.
ℹ This can happen when ggplot fails to infer the correct grouping structure in
the data.
ℹ Did you forget to specify a group aesthetic or to convert a numerical
variable into a factor?

The se=FALSE at the end is optional; if you omit it, you get that “envelope” around the line, which is fine here.

Note that with the labelling done this way, you can easily identify which roller-coaster is which.

The warning seems to be ggplot being over-zealous; the geom_point and the geom_smooth don’t need a label, but geom_text_repel certainly does. If it bothers you, move the label into the geom_text_repel:

ggplot(coasters, aes(x = drop, y = duration)) +
geom_point() + geom_text_repel(aes(label = coaster_name)) +
geom_smooth(method = "lm", se = FALSE)
geom_smooth() using formula = 'y ~ x'

$$\blacksquare$$

1. Would you say that roller-coasters with a larger drop tend to have a longer ride? Explain briefly.

Solution

I think there are two good answers here: “yes” and “kind of”. Supporting “yes” is the fact that the regression line does go uphill, so that overall, or on average, roller-coasters with a larger drop do tend to have a longer duration of ride as well. Supporting “kind of” is the fact that, though the regression line goes uphill, there are a lot of roller-coasters that are some way off the trend, far from the regression line. I am happy to go with either of those. I could also go with “not really” and the same discussion that I attached to “kind of”.

$$\blacksquare$$

1. Find a roller-coaster that is unusual compared to the others. What about its combination of drop and duration is unusual?

Solution

This is an invitation to find a point that is a long way off the line. I think the obvious choice is my first one below, but I would take either of the others as well:

• “Nitro” is a long way above the line. That means it has a long duration, relative to its drop. There are two other roller-coasters that have a larger drop but not as long a duration. In other words, this roller-coaster drops slowly, presumably by doing a lot of twisting, loop-the-loop and so on.

• “The Beast” is a long way below the line, so it has a short duration relative to its drop. It is actually the shortest ride of all, but is only a bit below average in terms of drop. This suggests that The Beast is one of those rides that drops a long way quickly.

• “Millennium Force” has the biggest drop of all, but a shorter-than-average duration. This looks like another ride with a big drop in it.

A roller-coaster that is “unusual” will have a residual that is large in size (either positive, like Nitro, or negative, like the other two). I didn’t ask you to find the residuals, but if you want to, augment from broom is the smoothest way to go:

library(broom)
duration.1 <- lm(duration ~ drop, data = coasters)
augment(duration.1, coasters) %>%
select(coaster_name, duration, drop, .resid) %>%
arrange(desc(abs(.resid)))

augment produces a data frame (of the original data frame with some new columns that come from the regression), so I can feed it into a pipe to do things with it, like only displaying the columns I want, and arranging them in order by absolute value of residual, so that the roller-coasters further from the line come out first. This identifies the three that we found above. The fourth one, “Ghost Rider”, is like Nitro in that it takes a (relatively) long time to fall not very far. You can also put augment in the middle of a pipe. What you may have to do then is supply the original data frame name to augment so that you have everything:

coasters %>%
lm(duration ~ drop, data = .) %>%
augment(coasters) %>%
arrange(desc(abs(.resid)))

I wanted to hang on to the roller-coaster names, so I added the data frame name to augment. If you don’t (that is, you just put augment() in the middle of a pipe), then augment “attempts to reconstruct the data from the model”.22 That means you wouldn’t get everything from the original data frame; you would just get the things that were in the regression. In this case, that means you would lose the coaster names.

A technicality (but one that you should probably care about): augment takes up to two inputs: a fitted model object like my duration.1, and an optional data frame to include other things from, like the coaster names. I had only one input to it in the pipe because the implied first input was the output from the lm, which doesn’t have a name; the input coasters in the pipe was what would normally be the second input to augment.

$$\blacksquare$$

## 18.24 Running and blood sugar

A diabetic wants to know how aerobic exercise affects his blood sugar. When his blood sugar reaches 170 (mg/dl), he goes out for a run at a pace of 10 minutes per mile. He runs different distances on different days. Each time he runs, he measures his blood sugar after the run. (The preferred blood sugar level is between 80 and 120 on this scale.) The data are in the file link. Our aim is to predict blood sugar from distance.

1. Read in the data and display the data frame that you read in.

Solution

From the URL is easiest. These are delimited by one space, as you can tell by looking at the file:

my_url <- "http://ritsokiguess.site/datafiles/runner.txt"
runs <- read_delim(my_url, " ")
Rows: 12 Columns: 2
── Column specification ────────────────────────────────────────────────────────
Delimiter: " "
dbl (2): distance, blood_sugar

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
runs

That looks like my data file.

$$\blacksquare$$

1. Make a scatterplot and add a smooth trend to it.

Solution

ggplot(runs, aes(x = distance, y = blood_sugar)) + geom_point() +
geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

blood_sugar should be on the vertical axis, since this is what we are trying to predict. Getting the x and the y right is easy on these, because they are the $$x$$ and $$y$$ for your plot.

$$\blacksquare$$

1. Would you say that the relationship between blood sugar and running distance is approximately linear, or not? It is therefore reasonable to use a regression of blood sugar on distance? Explain briefly.

Solution

I’d say that this is about as linear as you could ever wish for. Neither the pattern of points nor the smooth trend have any kind of noticeable bend in them. (Observing a lack of curvature in either the points or the smooth trend is enough.) The trend is a linear one, so using a regression will be just fine. (If it weren’t, the rest of the question would be kind of dumb.)

$$\blacksquare$$

1. Fit a suitable regression, and obtain the regression output.

Solution

Two steps: lm and then summary:

runs.1 <- lm(blood_sugar ~ distance, data = runs)
summary(runs.1)

Call:
lm(formula = blood_sugar ~ distance, data = runs)

Residuals:
Min      1Q  Median      3Q     Max
-7.8238 -3.6167  0.8333  4.0190  5.5476

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)  191.624      5.439   35.23 8.05e-12 ***
distance     -25.371      1.618  -15.68 2.29e-08 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 4.788 on 10 degrees of freedom
Multiple R-squared:  0.9609,    Adjusted R-squared:  0.957
F-statistic: 245.7 on 1 and 10 DF,  p-value: 2.287e-08

$$\blacksquare$$

1. How would you interpret the slope? That is, what is the slope, and what does that mean about blood sugar and running distance?

Solution

The slope is $$-25.37$$. This means that for each additional mile run, the runner’s blood sugar will decrease on average by about 25 units.

You can check this from the scatterplot. For example, from 2 to 3 miles, average blood sugar decreases from about 140 to about 115, a drop of 25.

$$\blacksquare$$

1. Is there a (statistically) significant relationship between running distance and blood sugar? How do you know? Do you find this surprising, given what you have seen so far? Explain briefly.

Solution

Look at the P-value either on the distance line (for its $$t$$-test) or for the $$F$$-statistic on the bottom line. These are the same: 0.000000023. (They will be the same any time there is one $$x$$-variable.) This P-value is way smaller than 0.05, so there is a significant relationship between running distance and blood sugar. This does not surprise me in the slightest, because the trend on the scatterplot is so clear, there’s no way it could have happened by chance if in fact there were no relationship between running distance and blood sugar.

$$\blacksquare$$

1. This diabetic is planning to go for a 3-mile run tomorrow and a 5-mile run the day after. Obtain suitable 95% intervals that say what his blood sugar might be after each of these runs.

Solution

This is a prediction interval, in each case, since we are talking about individual runs of 3 miles and 5 miles (not the mean blood sugar after all runs of 3 miles, which is what a confidence interval for the mean response would be). The procedure is to set up a data frame with the two distance values in it, and then feed that and the regression object into predict, coming up in a moment.

dists <- c(3, 5)
new <- tibble(distance = dists)
new

The important thing is that the name of the column of the new data frame must be exactly the same as the name of the explanatory variable in the regression. If they don’t match, predict won’t work. At least, it won’t work properly.23

If your first thought is datagrid, well, that will also work:

new2 <- datagrid(model = runs.1, distance = c(5, 10))
new2

Use whichever of these methods comes to your mind.

Then, predict, because you want prediction intervals rather than confidence intervals for the mean response (which is what marginaleffects gives you):

pp <- predict(runs.1, new, interval = "p")
pp
        fit       lwr       upr
1 115.50952 104.37000 126.64905
2  64.76667  51.99545  77.53788

and display this with the distances by the side:

cbind(new, pp)

or

data.frame(new, pp)

Blood sugar after a 3-mile run is predicted to be between 104 and 127; after a 5-mile run it is predicted to be between 52 and 77.5.

Extra: both cbind and data.frame are “base R” ways of combining a data frame with something else to make a new data frame. They are not from the tidyverse. The tidyverse way is via tibble or bind_cols, but they are a bit more particular about what they will take: tibble takes vectors (single variables) and bind_cols takes vectors or data frames. The problem here is that pp is not either of those:

class(pp)
[1] "matrix" "array" 

so that we have to use as_tibble first to turn it into a data frame, and thus:

pp %>% as_tibble() %>% bind_cols(new)

which puts things backwards, unless you do it like this:

new %>% bind_cols(as_tibble(pp))

which is a pretty result from very ugly code.

I also remembered that if you finish with a select, you get the columns in the order they were in the select:

pp %>%
as_tibble() %>%
bind_cols(new) %>%
select(c(distance, everything()))

everything is a so-called “select helper”. It means “everything except any columns you already named”, so this whole thing has the effect of listing the columns with distance first and all the other columns afterwards, in the order that they were in before.

$$\blacksquare$$

1. Which of your two intervals is longer? Does this make sense? Explain briefly.

Solution

The intervals are about 22.25 and 25.5 units long. The one for a 5-mile run is a bit longer. I think this makes sense because 3 miles is close to the average run distance, so there is a lot of “nearby” data. 5 miles is actually longer than any of the runs that were actually done (and therefore we are actually extrapolating), but the important point for the prediction interval is that there is less nearby data: those 2-mile runs don’t help so much in predicting blood sugar after a 5-mile run. (They help some, because the trend is so linear. This is why the 5-mile interval is not so much longer. If the trend were less clear, the 5-mile interval would be more noticeably worse.)

$$\blacksquare$$

## 18.25 Calories and fat in pizza

The file at link came from a spreadsheet of information about 24 brands of pizza: specifically, per 5-ounce serving, the number of calories, the grams of fat, and the cost (in US dollars). The names of the pizza brands are quite long. This file may open in a spreadsheet when you browse to the link, depending on your computer’s setup.

1. Read in the data and display at least some of the data frame. Are the variables of the right types? (In particular, why is the number of calories labelled one way and the cost labelled a different way?)

Solution

read_csv is the thing this time:

my_url <- "http://ritsokiguess.site/datafiles/Pizza.csv"
pizza <- read_csv(my_url)
Rows: 24 Columns: 4
── Column specification ────────────────────────────────────────────────────────
Delimiter: ","
chr (1): Type
dbl (3): Calories, Fat, Cost

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
pizza

The four variables are: the brand of pizza, which got read in as text, the number of calories (an integer), and the fat and cost, which are both decimal numbers so they get labelled dbl, which is short for “double-precision floating point number”.

Anyway, these are apparently the right thing.

Extra: I wanted to mention something else that I discovered yesterday.24 There is a package called rio that will read (and write) data in a whole bunch of different formats in a unified way.25 Anyway, the usual installation thing, done once:

install.packages("rio")

which takes a moment since it probably has to install some other packages too, and then you read in a file like this:

library(rio)
pizza3 <- import(my_url)
head(pizza3)

import figures that you have a .csv file, so it calls up read_csv or similar.

Technical note: rio does not use the read_ functions, so what it gives you is actually a data.frame rather than a tibble, so that when you display it, you get the whole thing even if it is long. Hence the head here and below to display the first six lines.

I originally had the data as an Excel spreadsheet, but import will gobble up that pizza too:

my_other_url <- "http://ritsokiguess.site/datafiles/Pizza_E29.xls"
pizza4 <- import(my_other_url)
head(pizza4)

The corresponding function for writing a data frame to a file in the right format is, predictably enough, called export.

$$\blacksquare$$

1. Make a scatterplot for predicting calories from the number of grams of fat. Add a smooth trend. What kind of relationship do you see, if any?

Solution

ggplot(pizza, aes(x = Fat, y = Calories)) + geom_point() +
geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

There is definitely an upward trend: the more fat, the more calories. The trend is more or less linear (or, a little bit curved: say what you like, as long as it’s not obviously crazy). I think, with this much scatter, there’s no real justification for fitting a curve.

$$\blacksquare$$

1. Fit a straight-line relationship, and display the intercept, slope, R-squared, etc. Is there a real relationship between the two variables, or is any apparent trend just chance?

Solution

lm, with summary:

pizza.1 <- lm(Calories ~ Fat, data = pizza)
summary(pizza.1)

Call:
lm(formula = Calories ~ Fat, data = pizza)

Residuals:
Min     1Q Median     3Q    Max
-55.44 -11.67   6.18  17.87  41.61

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)  194.747     21.605   9.014 7.71e-09 ***
Fat           10.050      1.558   6.449 1.73e-06 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 25.79 on 22 degrees of freedom
Multiple R-squared:  0.654, Adjusted R-squared:  0.6383
F-statistic: 41.59 on 1 and 22 DF,  p-value: 1.731e-06

To assess whether this trend is real or just chance, look at the P-value on the end of the Fat line, or on the bottom line where the $$F$$-statistic is (they are the same value of $$1.73\times 10^{-6}$$ or 0.0000017, so you can pick either). This P-value is really small, so the slope is definitely not zero, and therefore there really is a relationship between the two variables.

$$\blacksquare$$

1. Obtain a plot of the residuals against the fitted values for this regression. Does this indicate that there are any problems with this regression, or not? Explain briefly.

Solution

Use the regression object pizza.1:

ggplot(pizza.1, aes(x = .fitted, y = .resid)) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

On my residual plot, I see a slight curve in the smooth trend, but I am not worried about that because the residuals on the plot are all over the place in a seemingly random pattern (the grey envelope is wide and that is pretty close to going straight across). So I think a straight line model is satisfactory.

That’s all you needed, but it is also worth looking at a normal quantile plot of the residuals:

ggplot(pizza.1, aes(sample = .resid)) + stat_qq() + stat_qq_line()

A bit skewed to the left (the low ones are too low).

Also a plot of the absolute residuals, for assessing fan-out:

ggplot(pizza.1, aes(x = .fitted, y = abs(.resid))) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

A tiny bit of fan-in (residuals getting smaller in size as the fitted value gets bigger), but nothing much, I think.

Another way of assessing curvedness is to fit a squared term anyway, and see whether it is significant:

pizza.2 <- update(pizza.1, . ~ . + I(Fat^2))
summary(pizza.2)

Call:
lm(formula = Calories ~ Fat + I(Fat^2), data = pizza)

Residuals:
Min      1Q  Median      3Q     Max
-62.103 -14.280   5.513  15.423  35.474

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)  90.2544    77.8156   1.160   0.2591
Fat          25.9717    11.5121   2.256   0.0349 *
I(Fat^2)     -0.5702     0.4086  -1.395   0.1775
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 25.25 on 21 degrees of freedom
Multiple R-squared:  0.6834,    Adjusted R-squared:  0.6532
F-statistic: 22.66 on 2 and 21 DF,  p-value: 5.698e-06

The fat-squared term is not significant, so that curve on the smooth trend in the (first) residual plot was indeed nothing to get excited about.

$$\blacksquare$$

1. The research assistant in this study returns with two new brands of pizza (ones that were not in the original data). The fat content of a 5-ounce serving was 12 grams for the first brand and 20 grams for the second brand. For each of these brands of pizza, obtain a suitable 95% interval for the number of calories contained in a 5-ounce serving.

Solution

The suitable interval here is a prediction interval, because we are interested in each case in the calorie content of the particular pizza brands that the research assistant returned with (and not, for example, in the mean calorie content for all brands of pizza that have 12 grams of fat per serving). Thus:

newfat <- c(12, 20)
new <- tibble(Fat = newfat)
new
preds <- predict(pizza.1, new, interval = "p")
cbind(new, preds)

Use datagrid to make new if you like, but it is a very simple dataframe, so there is no obligation to do it that way.

Or, if you like:

as_tibble(preds) %>% bind_cols(new) %>% select(Fat, everything())

For the pizza with 12 grams of fat, the predicted calories are between 261 and 370 with 95% confidence, and for the pizza with 20 grams of fat, the calories are predicted to be between 337 and 454. (You should write down what these intervals are, and not leave the reader to find them in the output.)

(Remember the steps: create a new data frame containing the values to predict for, and then feed that into predict along with the model that you want to use to predict with. The variable in the data frame has to be called precisely Fat with a capital F, otherwise it won’t work.)

These intervals are both pretty awful: you get a very weak picture of how many calories per serving the pizza brands in question might contain. This is for two reasons: (i) there was a fair bit of scatter in the original relationship, R-squared being around 65%, and (ii) even if we knew perfectly where the line went (which we don’t), there’s no guarantee that individual brands of pizza would be on it anyway. (Prediction intervals are always hit by this double whammy, in that individual observations suffer from variability in where the line goes and variability around whatever the line is.)

I was expecting, when I put together this question, that the 20-grams-of-fat interval would be noticeably worse, because 20 is farther away from the mean fat content of all the brands. But there isn’t much to choose. For the confidence intervals for the mean calories of all brands with these fat contents, the picture is clearer:

plot_predictions(pizza.1, condition = "Fat")

A fat value of 12 is close to the middle of the data, so the interval is shorter, but a value of 20 is out near the extreme and the interval is noticeably longer.

This part was a fair bit of work for 3 points, so I’m not insisting that you explain your choice of a prediction interval over a confidence interval, but I think it is still a smart thing to do, even purely from a marks point of view, because if you get it wrong for a semi-plausible reason, you might pick up some partial credit. Not pulling out your prediction intervals from your output is a sure way to lose a point, however.

$$\blacksquare$$

## 18.26 Where should the fire stations be?

In city planning, one major issue is where to locate fire stations. If a city has too many fire stations, it will spend too much on running them, but if it has too few, there may be unnecessary fire damage because the fire trucks take too long to get to the fire.

The first part of a study of this kind of issue is to understand the relationship between the distance from the fire station (measured in miles in our data set) and the amount of fire damage caused (measured in thousands of dollars). A city recorded the fire damage and distance from fire station for 15 residential fires (which you can take as a sample of “all possible residential fires in that city”). The data are in link.

1. Read in and display the data, verifying that you have the right number of rows and the right columns.

Solution

A quick check of the data reveals that the data values are separated by exactly one space, so:

my_url <- "http://ritsokiguess.site/datafiles/fire_damage.txt"
fire <- read_delim(my_url, " ")
Rows: 15 Columns: 2
── Column specification ────────────────────────────────────────────────────────
Delimiter: " "
dbl (2): distance, damage

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
fire

15 observations (rows), and promised, and a column each of distances and amounts of fire damage, also as promised.

$$\blacksquare$$

1. * Obtain a 95% confidence interval for the mean fire damage. (There is nothing here from STAD29, and your answer should have nothing to do with distance.)

Solution

I wanted to dissuade you from thinking too hard here. It’s just an ordinary one-sample $$t$$-test, extracting the interval from it:

t.test(fire$damage)  One Sample t-test data: fire$damage
t = 12.678, df = 14, p-value = 4.605e-09
alternative hypothesis: true mean is not equal to 0
95 percent confidence interval:
21.94488 30.88178
sample estimates:
mean of x
26.41333 

Or

with(fire, t.test(damage))

One Sample t-test

data:  damage
t = 12.678, df = 14, p-value = 4.605e-09
alternative hypothesis: true mean is not equal to 0
95 percent confidence interval:
21.94488 30.88178
sample estimates:
mean of x
26.41333 

Ignore the P-value (it’s testing that the mean is the default zero, which makes no sense). The confidence interval either way goes from 21.9 to 30.9 (thousand dollars).

$$\blacksquare$$

1. Draw a scatterplot for predicting the amount of fire damage from the distance from the fire station. Add a smooth trend to your plot.

Solution

We are predicting fire damage, so that goes on the $$y$$-axis:

ggplot(fire, aes(x = distance, y = damage)) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

$$\blacksquare$$

1. * Is there a relationship between distance from fire station and fire damage? Is it linear or definitely curved? How strong is it? Explain briefly.

Solution

When the distance is larger, the fire damage is definitely larger, so there is clearly a relationship. I would call this one approximately linear: it wiggles a bit, but it is not to my mind obviously curved. I would also call it a strong relationship, since the points are close to the smooth trend.

$$\blacksquare$$

1. Fit a regression predicting fire damage from distance. How is the R-squared consistent (or inconsistent) with your answer from part~(here)?

Solution

The regression is an ordinary lm:

damage.1 <- lm(damage ~ distance, data = fire)
summary(damage.1)

Call:
lm(formula = damage ~ distance, data = fire)

Residuals:
Min      1Q  Median      3Q     Max
-3.4682 -1.4705 -0.1311  1.7915  3.3915

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)  10.2779     1.4203   7.237 6.59e-06 ***
distance      4.9193     0.3927  12.525 1.25e-08 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 2.316 on 13 degrees of freedom
Multiple R-squared:  0.9235,    Adjusted R-squared:  0.9176
F-statistic: 156.9 on 1 and 13 DF,  p-value: 1.248e-08

We need to display the results, since we need to see the R-squared in order to say something about it.

R-squared is about 92%, high, indicating a strong and linear relationship. Back in part~(here), I said that the relationship is linear and strong, which is entirely consistent with such an R-squared. (If you said something different previously, say how it does or doesn’t square with this kind of R-squared value.)

Points: one for fitting the regression, one for displaying it, and two (at the grader’s discretion) for saying what the R-squared is and how it’s consistent (or not) with part~(here).

Extra: if you thought the trend was “definitely curved”, you would find that a parabola (or some other kind of curve) was definitely better than a straight line. Here’s the parabola:

damage.2 <- lm(damage ~ distance + I(distance^2), data = fire)
summary(damage.2)

Call:
lm(formula = damage ~ distance + I(distance^2), data = fire)

Residuals:
Min      1Q  Median      3Q     Max
-2.8856 -1.6915 -0.0179  1.5490  3.6278

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)    13.3395     2.5303   5.272 0.000197 ***
distance        2.6400     1.6302   1.619 0.131327
I(distance^2)   0.3376     0.2349   1.437 0.176215
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 2.227 on 12 degrees of freedom
Multiple R-squared:  0.9347,    Adjusted R-squared:  0.9238
F-statistic: 85.91 on 2 and 12 DF,  p-value: 7.742e-08

There’s no evidence here that a quadratic is better.

Or you might even have thought from the wiggles that it was more like cubic:

damage.3 <- update(damage.2, . ~ . + I(distance^3))
summary(damage.3)

Call:
lm(formula = damage ~ distance + I(distance^2) + I(distance^3),
data = fire)

Residuals:
Min      1Q  Median      3Q     Max
-3.2325 -1.8377  0.0322  1.1512  3.1806

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)    10.8466     4.3618   2.487   0.0302 *
distance        5.9555     4.9610   1.200   0.2552
I(distance^2)  -0.8141     1.6409  -0.496   0.6296
I(distance^3)   0.1141     0.1608   0.709   0.4928
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 2.274 on 11 degrees of freedom
Multiple R-squared:  0.9376,    Adjusted R-squared:  0.9205
F-statistic: 55.07 on 3 and 11 DF,  p-value: 6.507e-07

No evidence that a cubic is better; that increase in R-squared up to about 94% is just chance (bearing in mind that adding any $$x$$, even a useless one, will increase R-squared).

How bendy is the cubic?

ggplot(fire, aes(x = distance, y = damage)) + geom_point() +
geom_smooth(method = "lm") +
geom_line(data = damage.3, aes(y = .fitted), colour = "red")
geom_smooth() using formula = 'y ~ x'

The cubic, in red, does bend a little, but it doesn’t do an obvious job of going through the points better than the straight line does. It seems to be mostly swayed by that one observation with damage over 40, and choosing a relationship by how well it fits one point is flimsy at the best of times. So, by Occam’s Razor, we go with the line rather than the cubic because it (i) fits equally well, (ii) is simpler.

$$\blacksquare$$

1. Obtain a 95% confidence interval for the mean fire damage for a residence that is 4 miles from the nearest fire station*. (Note the contrast with part~(here).)

Solution

This is a confidence interval for a mean response at a given value of the explanatory variable. This is as opposed to part~(here), which is averaged over all distances. So, follow the steps. Make a tiny data frame with this one value of distance:

new <- datagrid(model = damage.1, distance = 4)
new

and then

pp <- cbind(predictions(damage.1, newdata = new))
pp

28.5 to 31.4 (thousand dollars).

(I saved this one because I want to refer to it again later.)

$$\blacksquare$$

1. Compare the confidence intervals of parts (here) and (here). Specifically, compare their centres and their lengths, and explain briefly why the results make sense.

Solution

Let me just put them side by side for ease of comparison: part~(here) is:

t.test(fire$damage)  One Sample t-test data: fire$damage
t = 12.678, df = 14, p-value = 4.605e-09
alternative hypothesis: true mean is not equal to 0
95 percent confidence interval:
21.94488 30.88178
sample estimates:
mean of x
26.41333 

and part~(here)’s is

pp

The centre of the interval is higher for the mean damage when the distance is 4. This is because the mean distance is a bit less than 4:

fire %>% summarize(m = mean(distance))

We know it’s an upward trend, so our best guess at the mean damage is higher if the mean distance is higher (in (here), the distance is always 4: we’re looking at the mean fire damage for all residences that are 4 miles from a fire station.)

What about the lengths of the intervals? The one in (here) is about $$30.9-21.9=9$$ (thousand dollars) long, but the one in (here) is only $$31.4-28.5=2.9$$ long, much shorter. This makes sense because the relationship is a strong one: knowing the distance from the fire station is very useful, because the bigger it is, the bigger the damage going to be, with near certainty. Said differently, if you know the distance, you can estimate the damage accurately. If you don’t know the distance (as is the case in (here)), you’re averaging over a lot of different distances and thus there is a lot of uncertainty in the amount of fire damage also.

If you have some reasonable discussion of the reason why the centres and lengths of the intervals differ, I’m happy. It doesn’t have to be the same as mine.

$$\blacksquare$$

## 18.27 Making it stop

If you are driving, and you hit the brakes, how far do you travel before coming to a complete stop? Presumably this depends on how fast you are going. Knowing this relationship is important in setting speed limits on roads. For example, on a very bendy road, the speed limit needs to be low, because you cannot see very far ahead, and there could be something just out of sight that you need to stop for.

Data were collected for a typical car and driver, as shown in http://ritsokiguess.site/datafiles/stopping.csv. These are American data, so the speeds are miles per hour and the stopping distances are in feet.

1. Read in and display (probably all of) the data.

Solution

The usual:

my_url <- "http://ritsokiguess.site/datafiles/stopping.csv"
stopping <- read_csv(my_url)
Rows: 8 Columns: 2
── Column specification ────────────────────────────────────────────────────────
Delimiter: ","
dbl (2): speed, distance

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
stopping

There are only eight observations.

$$\blacksquare$$

1. Make a suitable plot of the data.

Solution

Two quantitative variables means a scatterplot. Stopping distance is the outcome, so that goes on the $$y$$-axis:

ggplot(stopping, aes(x=speed, y=distance)) + geom_point()

$$\blacksquare$$

1. Describe any trend you see in your graph.

Solution

It’s an upward trend, but not linear: the stopping distance seems to increase faster at higher speeds.

$$\blacksquare$$

1. Fit a linear regression predicting stopping distance from speed. (You might have some misgivings about doing this, but do it anyway.)

Solution

Having observed a curved relationship, it seems odd to fit a straight line. But we are going to do it anyway and then critique what we have:

stopping.1 <- lm(distance~speed, data=stopping)
summary(stopping.1)

Call:
lm(formula = distance ~ speed, data = stopping)

Residuals:
Min      1Q  Median      3Q     Max
-32.738 -22.351  -7.738  16.622  47.083

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) -44.1667    22.0821   -2.00   0.0924 .
speed         5.6726     0.5279   10.75 3.84e-05 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 34.21 on 6 degrees of freedom
Multiple R-squared:  0.9506,    Adjusted R-squared:  0.9424
F-statistic: 115.5 on 1 and 6 DF,  p-value: 3.837e-05

Extra: note that R-squared is actually really high. We come back to that later.

$$\blacksquare$$

1. Plot the residuals against the fitted values for this regression.

Solution

ggplot(stopping.1, aes(x=.fitted, y=.resid)) + geom_point()

$$\blacksquare$$

1. What do you learn from the residual plot? Does that surprise you? Explain briefly.

Solution

The points on the residual plot form a (perfect) curve, so the original relationship was a curve. This is exactly what we saw on the scatterplot, so to me at least, this is no surprise.

(Make sure you say how you know that the original relationship was a curve from looking at the residual plot. Joined-up thinking. There are two ways we know that the relationship is a curve. Get them both.)

$$\blacksquare$$

1. What is the actual relationship between stopping distance and speed, according to the physics? See if you can find out. Cite any books or websites that you use: that is, include a link to a website, or give enough information about a book that the grader could find it.

Solution

I searched for “braking distance and speed” and found the first two things below, that seemed to be useful. Later, I was thinking about the fourth point (which came out of my head) and while searching for other things about that, I found the third thing:

• a British road safety website, that says “The braking distance depends on how fast the vehicle was travelling before the brakes were applied, and is proportional to the square of the initial speed.”
• the Wikipedia article on braking distance, which gives the actual formula. This is the velocity squared, divided by a constant that depends on the coefficient of friction. (That is why your driving instructor tells you to leave a bigger gap behind the vehicle in front if it is raining, and an even bigger gap if it is icy.)
• an Australian math booklet that talks specifically about braking distance and derives the formula (and the other equations of motion).
• also, if you have done physics, you might remember the equation of motion $$v^2 = u^2 + 2as$$, where $$u$$ is the initial velocity, $$v$$ is the final velocity, $$a$$ is the acceleration and $$s$$ is the distance covered. In this case, $$v=0$$ (the car is stationary at the end), and so $$-u^2/2a = s$$. The acceleration is negative (the car is slowing down), so the left side is, despite appearances, positive. There seems to be a standard assumption that deceleration due to braking is constant (the same for all speeds), at least if you are trying to stop a car in a hurry.

These are all saying that we should add a speed-squared term to our regression, and then we will have the relationship exactly right, according to the physics.

Extra: Another way to measure how far you are behind the vehicle in front is time. Many of the British “motorways” (think 400-series highways) were built when I was young, and I remember a TV commercial that said “Only a Fool Breaks the Two Second Rule”.26 In those days (the linked one is from the 1970s),27 a lot of British drivers were not used to going that fast, or on roads that straight, so this was a way to know how big a gap to leave, so that you had time to take evasive action if needed. The value of the two-second rule is that it works for any speed, and you don’t have to remember a bunch of stopping distances. (When I did my (Canadian) driving theory test, I think I worked out and learned a formula for the stopping distances that I could calculate in my head. I didn’t have to get very close since the test was multiple-choice.)

$$\blacksquare$$

1. Fit the relationship that your research indicated (in the previous part) and display the results. Comment briefly on the R-squared value.

Solution

Add a squared term in speed:

stopping.2 <- lm(distance~speed+I(speed^2), data=stopping)
summary(stopping.2)

Call:
lm(formula = distance ~ speed + I(speed^2), data = stopping)

Residuals:
1        2        3        4        5        6        7        8
-1.04167  0.98214  0.08929  1.27976 -0.44643 -0.08929 -2.64881  1.87500

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) 1.041667   1.429997   0.728    0.499
speed       1.151786   0.095433  12.069 6.89e-05 ***
I(speed^2)  0.064583   0.001311  49.267 6.51e-08 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 1.699 on 5 degrees of freedom
Multiple R-squared:  0.9999,    Adjusted R-squared:  0.9999
F-statistic: 2.462e+04 on 2 and 5 DF,  p-value: 1.039e-10

The R-squared now is basically 1, so that the model fits very close to perfectly.

Extra: you probably found in your research that the distance should be just something times speed squared, with no constant or linear term. Here, though, we have a significant linear term as well. That is probably just chance, since the distances in the data look as if they have been rounded off. With more accurate values, I think the linear term would have been closer to zero.

If you want to go literally for the something-times-speed-squared, you can do that. This doesn’t quite work:

stopping.3x <- lm(distance~I(speed^2), data=stopping)
summary(stopping.3x)

Call:
lm(formula = distance ~ I(speed^2), data = stopping)

Residuals:
Min       1Q   Median       3Q      Max
-14.7327  -3.4670   0.6761   6.2323   8.4513

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept) 14.732704   4.362859   3.377   0.0149 *
I(speed^2)   0.079796   0.001805  44.218 8.96e-09 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 8.514 on 6 degrees of freedom
Multiple R-squared:  0.9969,    Adjusted R-squared:  0.9964
F-statistic:  1955 on 1 and 6 DF,  p-value: 8.958e-09

because it still has an intercept in it. In R, the intercept is denoted by 1. It is always included, unless you explicitly remove it. Some odd things start to happen if you remove the intercept, so it is not a good thing to do unless you know what you are doing. The answers here have some good discussion. Having decided that you are going to remove the intercept, you can remove it the same way as anything else (see update in the multiple regression lecture) with “minus”. I haven’t shown you this, so if you do it, you will need to cite your source: that is, say where you learned what to do:

stopping.3 <- lm(distance~I(speed^2)-1, data=stopping)
summary(stopping.3)

Call:
lm(formula = distance ~ I(speed^2) - 1, data = stopping)

Residuals:
Min       1Q   Median       3Q      Max
-12.6123  -0.7859  10.5314  15.5314  19.2141

Coefficients:
Estimate Std. Error t value Pr(>|t|)
I(speed^2) 0.084207   0.001963   42.89 9.77e-10 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 13.42 on 7 degrees of freedom
Multiple R-squared:  0.9962,    Adjusted R-squared:  0.9957
F-statistic:  1840 on 1 and 7 DF,  p-value: 9.772e-10

The R-squared is still extremely high, much higher than for the straight line. The coefficient value, as I said earlier (citing Wikipedia), depends on the coefficient of friction; the stopping distances you see typically are based on a dry road, so you have to allow extra distance (or time: see above) if the road is not dry.

$$\blacksquare$$

1. Somebody says to you “if you have a regression with a high R-squared, like 95%, there is no need to look for a better model.” How do you respond to this? Explain briefly.

Solution

An example of a regression with an R-squared of 95% is the straight-line fit from earlier in this question. This is an example of a regression that fits well but is not appropriate because it doesn’t capture the form of the relationship.

In general, we are saying that no matter how high R-squared is, we might still be able to improve on the model we have. The flip side is that we might not be able to do any better (with another data set) than an R-squared of, say, 30%, because there is a lot of variability that is, as best as we can assess it, random and not explainable by anything.

Using R-squared as a measure of absolute model quality is, thus, a mistake. Or, to say it perhaps more clearly, asking “how high does R-squared have to be to indicate a good fit?” is asking the wrong question. The right thing to do is to concentrate on getting the form of the model right, and thereby get the R-squared as high as we can for that data set (which might be very high, as here, or not high at all).

$$\blacksquare$$

## 18.28 Predicting height from foot length

Is it possible to estimate the height of a person from the length of their foot? To find out, 33 (male) students had their height and foot length measured. The data are in http://ritsokiguess.site/datafiles/heightfoot.csv.

1. Read in and display (some of) the data. (If you are having trouble, make sure you have exactly the right URL. The correct URL has no spaces or other strange characters in it.)

Solution

The usual:

my_url <- "http://ritsokiguess.site/datafiles/heightfoot.csv"
hf <- read_csv(my_url)
Rows: 33 Columns: 2
── Column specification ────────────────────────────────────────────────────────
Delimiter: ","
dbl (2): height, foot

ℹ Use spec() to retrieve the full column specification for this data.
ℹ Specify the column types or set show_col_types = FALSE to quiet this message.
hf

Call the data frame whatever you like, but keeping away from the names height and foot is probably wise, since those are the names of the columns.

There are indeed 33 rows as promised.

Extra: my comment in the question was to help you if you copy-pasted the file URL into R Studio. Depending on your setup, this might have gotten pasted with a space in it, at the point where it is split over two lines. The best way to proceed, one that won’t run into this problem, is to right-click on the URL and select Copy Link Address (or the equivalent on your system), and then it will put the whole URL on the clipboard in one piece, even if it is split over two lines in the original document, so that pasting it will work without problems.

$$\blacksquare$$

1. Make a suitable plot of the two variables in the data frame.

Solution

They are both quantitative, so a scatter plot is called for:

ggplot(hf, aes(y=height, x=foot)) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

I added a smooth trend, or you could just plot the points. (This is better than plotting a regression line at this stage, because we haven’t yet thought about whether the trend is straight.)

Now that we’ve seen the scatterplot, the trend looks more or less straight (but you should take a look at the scatterplot first, with or without smooth trend, before you put a regression line on it). That point top left is a concern, though, which brings us to…

$$\blacksquare$$

1. Are there any observations not on the trend of the other points? What is unusual about those observations?

Solution

The observation with height greater than 80 at the top of the graph looks like an outlier and does not follow the trend of the rest of the points. Or, this individual is much taller than you would expect for someone with a foot length of 27 inches. Or, this person is over 7 feet tall, which makes little sense as a height. Say something about what makes this person be off the trend.

$$\blacksquare$$

1. Fit a regression predicting height from foot length, including any observations that you identified in the previous part. For that regression, plot the residuals against the fitted values and make a normal quantile plot of the residuals.

Solution

These things. Displaying the summary of the regression is optional, but gives the grader an opportunity to check that your work is all right so far.

hf.1 <- lm(height~foot, data=hf)
summary(hf.1)

Call:
lm(formula = height ~ foot, data = hf)

Residuals:
Min      1Q  Median      3Q     Max
-4.7491 -1.3901 -0.0310  0.8918 12.9690

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)  34.3363     9.9541   3.449 0.001640 **
foot          1.3591     0.3581   3.795 0.000643 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 3.102 on 31 degrees of freedom
Multiple R-squared:  0.3173,    Adjusted R-squared:  0.2952
F-statistic: 14.41 on 1 and 31 DF,  p-value: 0.0006428
ggplot(hf.1, aes(x=.fitted, y=.resid)) + geom_point()
ggplot(hf.1, aes(sample=.resid)) + stat_qq() + stat_qq_line()

Note that we did not exclude the off-trend point. Removing points because they are outliers is a bad idea. This is a good discussion of the issues.

$$\blacksquare$$

1. Earlier, you identified one or more observations that were off the trend. How does this point or points show up on each of the plots you drew in the previous part?

Solution

On its own at the top in both cases; the large positive residual on the first plot, and the unusually large value at the top right of the normal quantile plot. (You need to say one thing about each graph, or say as I did that the same kind of thing happens on both graphs.)

Extra: in the residuals vs. fitted values, the other residuals show a slight upward trend. This is because the regression line for these data, with the outlier, is pulled (slightly) closer to the outlier and thus slightly further away from the other points, particularly the ones on the left, compared to the same data but with the outlier removed (which you will be seeing shortly). If the unusual point had happened to have an extreme $$x$$ (foot length) as well, the effect would have been more pronounced.

This is the kind of thing I mean (made-up data):

tibble(x = 1:10) %>%
mutate(y = rnorm(10, 10+2*x, 1)) %>%
mutate(y = ifelse(x == 9, 40, y)) -> madeup
madeup
ggplot(madeup, aes(x=x, y=y)) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

The second-last point is off a clearly linear trend otherwise (the smooth gets “distracted” by the outlying off-trend point). Fitting a regression anyway and looking at the residual plot gives this:

madeup.1 <- lm(y~x, data = madeup)
ggplot(madeup.1, aes(x = .fitted, y = .resid)) + geom_point()

This time you see a rather more obvious downward trend in the other residuals. The problem is not with them, but with the one very positive residual, corresponding to the outlier that is way off the trend on the scatterplot.

The residuals in a regression have to add up to zero. If one of them is very positive (as in the one you did and the example I just showed you), at least some of the other residuals have to become more negative to compensate – the ones on the right just above and the ones on the left in the one you did. If you have done STAC67, you will have some kind of sense of why that is: think about the two equations you have to solve to get the estimates of intercept and slope, and how they are related to the residuals. Slide 6 of this shows them; at the least squares estimates, these two partial derivatives both have to be zero, and the things inside the brackets are the residuals.

$$\blacksquare$$

1. Any data points that concerned you earlier were actually errors. Create and save a new data frame that does not contain any of those data points.

Solution

Find a way to not pick that outlying point. For example, you can choose the observations with height less than 80:

hf %>% filter(height<80) -> hfx
hfx

Only 32 rows left.

There are many other possibilities. Find one.

$$\blacksquare$$

1. Run a regression predicting height from foot length for your data set without errors. Obtain a plot of the residuals against fitted values and a normal quantile plot of the residuals for this regression.

Solution

Code-wise, the same as before, but with the new data set:

hf.2 <- lm(height~foot, data=hfx)
summary(hf.2)

Call:
lm(formula = height ~ foot, data = hfx)

Residuals:
Min      1Q  Median      3Q     Max
-4.5097 -1.0158  0.4757  1.1141  3.9951

Coefficients:
Estimate Std. Error t value Pr(>|t|)
(Intercept)  30.1502     6.5411   4.609 7.00e-05 ***
foot          1.4952     0.2351   6.360 5.12e-07 ***
---
Signif. codes:  0 '***' 0.001 '**' 0.01 '*' 0.05 '.' 0.1 ' ' 1

Residual standard error: 2.029 on 30 degrees of freedom
Multiple R-squared:  0.5741,    Adjusted R-squared:  0.5599
F-statistic: 40.45 on 1 and 30 DF,  p-value: 5.124e-07
ggplot(hf.2, aes(x=.fitted, y=.resid)) + geom_point()
ggplot(hf.2, aes(sample=.resid)) + stat_qq() + stat_qq_line()

$$\blacksquare$$

1. Do you see any problems on the plots you drew in the previous part? Explain briefly.

Solution

For myself, I see a random scatter of points on the first plot, and points close to the line on the second one. Thus I don’t see any problems at all. I would declare myself happy with the second regression, after removing the outlier. (Remember that we removed the outlier because it was an error, not just because it was an outlier. Outliers can be perfectly correct data points, and if they are, they have to be included in the modelling.)

You might have a different point of view, in which case you need to make the case for it. You might see a (very mild) case of fanning out in the first plot, or the two most negative residuals might be a bit too low. These are not really outliers, though, not at least in comparison to what we had before.

Extra: a standard diagnostic for fanning-out is to plot the absolute values of the residuals against the fitted values, with a smooth trend. If this looks like an increasing trend, there is fanning-out; a decreasing trend shows fanning-in. The idea is that we want to see whether the residuals are changing in size (for example, getting more positive and more negative both):

ggplot(hf.2, aes(x=.fitted, y=abs(.resid))) + geom_point() + geom_smooth()
geom_smooth() using method = 'loess' and formula = 'y ~ x'

No evidence of fanning-out at all. In fact, the residuals seem to be smallest in size in the middle.

Another thing you might think of is to try Box-Cox:

boxcox(height~foot, data=hfx)

It looks as if the best $$\lambda$$ is $$-1$$, and we should predict one over height from foot length. But this plot is deceiving, since it doesn’t even show the whole confidence interval for $$\lambda$$! We should zoom out (a lot) more:

boxcox(height~foot, data=hfx, lambda = seq(-8, 8, 0.1))

This shows that the confidence interval for $$\lambda$$ goes from about $$-7$$ to almost 5: that is, any value of $$\lambda$$ in that interval is supported by the data! This very definitely includes the do-nothing $$\lambda=1$$, so there is really no support for any kind of transformation.

$$\blacksquare$$

1. Find a way to plot the data and both regression lines on the same plot, in such a way that you can see which regression line is which. If you get help from anything outside the course materials, cite your source(s).

Solution

This is the same idea as with the windmill data, page 22, though this one is a bit easier since everything is linear (no curves).

The easiest way is to use geom_smooth twice, once with the original data set, and then on the one with the outlier removed:

ggplot(hf, aes(y=height, x=foot)) + geom_point() + geom_smooth(method = "lm", se=F) +
geom_smooth(data=hfx, method="lm", colour="red", se=F)
geom_smooth() using formula = 'y ~ x'
geom_smooth() using formula = 'y ~ x'

I would use the original data set as the “base”, since we want to plot its points (including the outlier) as well as its line. Then we want to plot just the line for the second data set. This entails using a data= in the second geom_smooth, to say that we want to get this regression line from a different data set, and also entails drawing this line in a different colour (or in some way distinguishing it from the first one). Putting the colour outside an aes is a way to make the whole line red. (Compare how you make points different colours according to their value on a third variable.)

This is, I think, the best way to do it. You can mimic the idea that I used for the windmill data:

ggplot(hf, aes(y=height, x=foot)) + geom_point() + geom_smooth(method = "lm", se=F) +
geom_line(data=hf.2, aes(y = .fitted))
geom_smooth() using formula = 'y ~ x'

but this is not as good, because you don’t need to use the trickery with geom_line: the second trend is another regression line not a curve, and we know how to draw regression lines with geom_smooth without having to actually fit them. (Doing it this way reveals that you are copying without thinking, instead of wondering whether there is a better way to do it.)

The idea of using a different data set in different “layers” of a plot is quite well-known. For example, the idea is the one in here, though used for a different purpose there (plotting different sets of points instead of different lines).

$$\blacksquare$$

1. Discuss briefly how removing the observation(s) that were errors has changed where the regression line goes, and whether that is what you expected.

Solution

The regression line for the original data (my blue line) is pulled up compared to the one with outliers removed (red).

This is not very surprising, because we know that regression lines tend to get pulled towards outliers. What was surprising to me was that the difference wasn’t very big. Even at the low end, where the lines differ the most, the difference in predicted height is only about one inch. Since regression lines are based on means, I would have expected a big outlier to have moved the line a lot more.

Say something about what you expected, and say something insightful about whether that was what you saw.

Extra: the regression lines are very similar, but their R-squared values are not: 32% and 57% respectively. Having a point far from the line has a big (downward) impact on R-squared.

$$\blacksquare$$

1. This is a base graphics graph rather than a ggplot one, but it will do for our purposes.↩︎

2. Roller-coasters work by gravity, so there must be some drop.↩︎

3. These are not to be confused with what your mom insists that you place between your coffee mug and the table.↩︎

4. If you had just one $$x$$, you’d use a $$t$$-test for its slope, and if you were testing all the $$x$$’s, you’d use the global $$F$$-test that appears in the regression output.↩︎

5. This is a base graphics graph rather than a ggplot one, but it will do for our purposes.↩︎

6. Recall that adding $$x$$-variables to a regression will always make R-squared go up, even if they are just random noise.↩︎

7. This is not, I don’t think, a real word, but I mean size emphasizing how big a boy is generally, rather than how small.↩︎

8. Correlations have to go up beyond 0.50 before they start looking at all interesting.↩︎

9. The suspicion being that we can, since the scatterplot suggested serious non-linearity.↩︎

10. Again, not a surprise, given our initial scatterplot.↩︎

11. Now we can use that word significant.↩︎

12. This might be overkill at this point, since we really only care about whether our data values are reasonable, and often just looking at the highest and lowest values will tell us that.↩︎

13. Mathematically, $$e^x$$ is approximately $$1+x$$ for small $$x$$, which winds up meaning that the slope in a model like this, if it is small, indicates about the percent increase in the response associated with a 1-unit change in the explanatory variable. Note that this only works with $$e^x$$ and natural logs, not base 10 logs or anything like that.↩︎

14. When this was graded, it was 3 marks, to clue you in that there are three things to say.↩︎

15. I have always used dots, but in the spirit of the tidyverse I suppose I should use underscores.↩︎

16. The summary output is more designed for looking at than for extracting things from.↩︎

17. These days, there are apps that will let you do this with your phone. I found one called Clinometer. See also link.↩︎

18. The very negative residuals are at the left and right of the residual plot; they are there because the relationship is a curve. If you were to look at the residuals from the model with length-squared, you probably wouldn’t see this.↩︎

19. The value, but throw away the minus sign if it has one.↩︎

20. Roller-coasters work by gravity, so there must be some drop.↩︎

21. These are not to be confused with what your mom insists that you place between your coffee mug and the table.↩︎

22. A quote from the package vignette.↩︎

23. It won’t give you an error, but it will go back to the original data frame to get distances to predict from, and you will get very confused. This is another example of (base) R trying to make life easier for you, but when it fails, it fails.↩︎

24. R is like that: sometimes it seems as if it has infinite depth.↩︎

25. It does this by figuring what kind of thing you have, from the extension to its filename, and then calling an appropriate function to read in or write out the data. This is an excellent example of “standing on the shoulders of giants” to make our lives easier. The software does the hard work of figuring out what kind of thing you have and how to read it in; all we do is say import.↩︎

26. This is perhaps not a commercial so much as a public safety message.↩︎

27. There are some typical British cars of the era in the commercial.↩︎