In [1]:
from dsc80_utils import *

Lecture 13 – Linear Regression¶

DSC 80, Spring 2024¶

Announcements 📣¶

  • Project 3 is due tomorrow.
  • The Final Project will be released over the weekend.
  • Lab 7 released, due Wednesday.

Agenda 📆¶

  • Modeling.
  • Case study: Restaurant tips 🧑‍🍳.
  • Regression in sklearn.

Conceptually, today will mostly be review from DSC 40A, but we'll introduce a few new practical tools that we'll build upon next week.

Midterm Survey Results¶

  • 107 responses out of 140 = 76%. No extra credit :(
  • Pluses:
    • Opting-in policy for lecture / discussion attendance
    • Extensions for assignments
    • q.dsc80.com for questions
    • Course topics are interesting
    • Including GenAI
    • Course staff are good!
  • Deltas:
    • Want more practice with exam-style questions
    • Lectures go fast
    • Lots of assignments, e.g. Lab + Midterm + Project in same week
    • Midterm was hard, maybe unfairly so!

Some Thoughts Going Forward¶

  • Yes, we will do a lot more practice problems during class!
    • And I will try to slow down lectures.
    • Feel free to ask me to repeat a slide / code cell via q.dsc80.com.
  • Final exam will be easier than the midterm -- midterm questions were harder than I intended.
  • Learning to read code is a very important skill that you should practice.

Question 🤔 (Answer at q.dsc80.com)

From the Fa23 final.

Consider the following corpus:

Document number Content
1               yesterday rainy today sunny
2               yesterday sunny today sunny
3               today rainy yesterday today
4               yesterday yesterday today today

Which words have a TF-IDF score of 0 for all four documents?

Question 🤔 (Answer at q.dsc80.com)

Taken from the Spring 2022 Final Exam.

The DataFrame below contains a corpus of four song titles, labeled from 0 to 3.

track_name
0 i hate you i love you i hate that i love you
1 love me like a love song
2 love you better
3 hate sosa

Part 1: What is the TF-IDF of the word "hate" in Song 0's title? Use base 2 in your logarithm, and give your answer as a simplified fraction.

Part 2: Which word in Song 0's title has the highest TF-IDF?

Part 3: Let $\text{tfidf}(t, d)$ be the TF-IDF of term $t$ in document $d$, and let $\text{bow}(t, d)$ be the number of occurrences of term $t$ in document $d$.

Select all correct answers below.

  • If $\text{tfidf}(t, d) = 0$, then $\text{bow}(t, d) = 0$.
  • If $\text{bow}(t, d) = 0$, then $\text{tfidf}(t, d) = 0$.
  • Neither of the above statements are necessarily true.

Modeling¶

Reflection¶

So far this quarter, we've learned how to:

  • Extract information from tabular data using pandas and regular expressions.

  • Clean data so that it best represents an underlying data generating process.

    • Missingness analyses and imputation.

  • Collect data from the internet through scraping and APIs, and parse it using BeautifulSoup.

  • Perform exploratory data analysis through aggregation, visualization, and the computation of summary statistics like TF-IDF.

  • Infer about the relationships between samples and populations through hypothesis and permutation testing.

  • Now, let's make predictions.

No description has been provided for this image

Modeling¶

  • A model is a set of assumptions about how data were generated.

  • George Box, a famous statistician, once said "All models are wrong, but some are useful." What did he mean?

Philosophy¶

"It has been said that "all models are wrong but some models are useful." In other words, any model is at best a useful fiction—there never was, or ever will be, an exactly normal distribution or an exact linear relationship. Nevertheless, enormous progress has been made by entertaining such fictions and using them as approximations."

"Since all models are wrong the scientist cannot obtain a "correct" one by excessive elaboration. On the contrary following William of Occam he should seek an economical description of natural phenomena. Just as the ability to devise simple but evocative models is the signature of the great scientist so overelaboration and overparameterization is often the mark of mediocrity."

Goals of modeling¶

  1. To make accurate predictions regarding unseen data.

    • Given this dataset of past UCSD data science students' salaries, can we predict your future salary? (regression)
    • Given this dataset of images, can we predict if this new image is of a dog, cat, or zebra? (classification)

  1. To make inferences about complex phenomena in nature.

    • Is there a linear relationship between the heights of children and the heights of their biological mothers?
    • The weights of smoking and non-smoking mothers' babies babies in my sample are different – how confident am I that this difference exists in the population?
No description has been provided for this image

  • Of the two focuses of models, we will focus on prediction.

  • In the above taxonomy, we will focus on supervised learning.

  • We'll start with regression before moving to classification.

Features¶

  • A feature is a measurable property of a phenomenon being observed.

    • Other terms for "feature" include "(explanatory) variable" and "attribute".
    • Typically, features are the inputs to models.

  • In DataFrames, features typically correspond to columns, while rows typically correspond to different individuals.

  • Some features come as part of a dataset, e.g. weight and height, but others we need to create given existing features, for example:

$$\text{BMI} = \frac{\text{weight (kg)}}{\text{[height (m)]}^2}$$
  • Example: TF-IDF creates features that summarize documents!

Example: Restaurant tips 🧑‍🍳¶

About the data¶

What features does the dataset contain? Is this likely a recent dataset, or an older one?

In [2]:
# The dataset is built into plotly!
tips = px.data.tips()
tips
Out[2]:
total_bill tip sex smoker day time size
0 16.99 1.01 Female No Sun Dinner 2
1 10.34 1.66 Male No Sun Dinner 3
2 21.01 3.50 Male No Sun Dinner 3
... ... ... ... ... ... ... ...
241 22.67 2.00 Male Yes Sat Dinner 2
242 17.82 1.75 Male No Sat Dinner 2
243 18.78 3.00 Female No Thur Dinner 2

244 rows × 7 columns

Predicting tips¶

  • Goal: Given various information about a table at a restaurant, we want to predict the tip that a server will earn.
  • Why might a server be interested in doing this?
    • To determine which tables are likely to tip the most (inference).
    • To predict earnings over the next month (prediction).

Exploratory data analysis¶

  • The most natural feature to look at first is total bills.
  • As such, we should explore the relationship between total bills and tips. Moving forward:
    • $x$: Total bills.
    • $y$: Tips.
In [3]:
fig = tips.plot(kind='scatter', x='total_bill', y='tip', title='Tip vs. Total Bill')
fig.update_layout(xaxis_title='Total Bill', yaxis_title='Tip')

Model #1: Constant¶

  • Let's start simple, by ignoring all features. Suppose our model assumes every tip is given by a constant dollar amount:
$$\text{tip} = h^{\text{true}}$$
  • Model: There is a single tip amount $h^{\text{true}}$ that all customers pay.
    • Correct? No!
    • Useful? Perhaps. An estimate of $h^{\text{true}}$, denoted by $h^*$, can allow us to predict future tips.
No description has been provided for this image

Estimating $h^{\text{true}}$¶

  • The true parameter $h^{\text{true}}$ is determined by the universe. We'll never get to see it, so we need to estimate it.
  • There are several ways we could estimate $h^{\text{true}}$. For instance, we could use domain knowledge: for instance, assume that everyone clicks the $1 tip option when buying coffee.
  • But, perhaps, we should look at the tips that we've already seen in our dataset to estimate $h^{\text{true}}$.

Looking at the data¶

Our estimate for $h^{\text{true}}$ should be a good summary statistic of the distribution of tips.

In [4]:
fig = tips.plot(kind='hist', x='tip', title='Distribution of Tip', nbins=20)
fig.update_layout(xaxis_title='Tip', yaxis_title='Frequency')

Empirical risk minimization¶

  • In DSC 40A, we established a framework for estimating model parameters:
    • Choose a loss function, which measures how "good" a single prediction is.
    • Minimize empirical risk, to find the best estimate for the dataset that we have.
  • Depending on which loss function we choose, we will end up with different $h^*$ (which are estimates of $h^{\text{true}})$.
  • If we choose squared loss, then our empirical risk is mean squared error:
$$\text{MSE} = \frac{1}{n} \sum_{i = 1}^n ( y_i - h )^2 \overset{\text{calculus}}\implies h^* = \text{mean}(y)$$
Remember, tips are our $y$ variable.
  • If we choose absolute loss, then our empirical risk is mean absolute error:
$$\text{MAE} = \frac{1}{n} \sum_{i = 1}^n | y_i - h | \overset{\text{algebra}}\implies h^* = \text{median}(y)$$

The mean tip¶

Let's suppose we choose squared loss, meaning that $h^* = \text{mean}(y)$.

In [5]:
mean_tip = tips['tip'].mean()
mean_tip
Out[5]:
2.9982786885245902

$h^* = 2.998$ is our fit model – it was fit to our training data (the data we have available to learn from).

Let's visualize this prediction.

In [6]:
fig = px.scatter(tips, x='total_bill', y='tip')
fig.add_hline(mean_tip, line_width=3, line_color='orange', opacity=1)
fig.update_layout(title='Tip vs. Total Bill',
                  xaxis_title='Total Bill', yaxis_title='Tip')

Note that to make predictions, this model ignores total bill (and all other features), and predicts the same tip for all tables.

The quality of predictions¶

  • Question: How can we quantify how good this constant prediction is at predicting tips in our training data – that is, the data we used to fit the model?
  • One answer: use the mean squared error. If $y_i$ represents the $i$th actual value and $H(x_i)$ represents the $i$th predicted value, then:
$$\text{MSE} = \frac{1}{n} \sum_{i = 1}^n \big( y_i - H(x_i) \big)^2$$
In [7]:
np.mean((tips['tip'] - mean_tip) ** 2)
Out[7]:
1.9066085124966428
In [8]:
# The same! A fact from 40A.
np.var(tips['tip'])
Out[8]:
1.9066085124966428
  • Issue: The units of MSE are "dollars squared", which are a little hard to interpret.

Root mean squared error¶

  • Often, to measure the quality of a regression model's predictions, we will use the root mean squared error (RMSE):
$$\text{RMSE} = \sqrt{\frac{1}{n} \sum_{i = 1}^n \big( y_i - H(x_i) \big)^2}$$
  • The units of the RMSE are the same as the units of the original $y$ values – dollars, in this case.
  • Important: Minimizing MSE is the same as minimizing RMSE; the constant tip $h^*$ that minimizes MSE is the same $h^*$ that minimizes RMSE. (Why?)

Computing and storing the RMSE¶

Since we'll compute the RMSE for our future models too, we'll define a function that can compute it for us.

In [9]:
def rmse(actual, pred):
    return np.sqrt(np.mean((actual - pred) ** 2))

Let's compute the RMSE of our constant tip's predictions, and store it in a dictionary that we can refer to later on.

In [10]:
rmse(tips['tip'], mean_tip)
Out[10]:
1.3807999538298958
In [11]:
rmse_dict = {}
rmse_dict['constant tip amount'] = rmse(tips['tip'], mean_tip)
rmse_dict
Out[11]:
{'constant tip amount': 1.3807999538298958}

Key idea: Since the mean minimizes RMSE for the constant model, it is impossible to change the mean_tip argument above to another number and yield a lower RMSE.

Question 🤔 (Answer at q.dsc80.com)

Remember, you can always ask questions at q.dsc80.com!

Model #2: Simple linear regression using total bill¶

  • We haven't yet used any of the features in the dataset. The first natural feature to look at is 'total_bill'.
In [12]:
tips.head()
Out[12]:
total_bill tip sex smoker day time size
0 16.99 1.01 Female No Sun Dinner 2
1 10.34 1.66 Male No Sun Dinner 3
2 21.01 3.50 Male No Sun Dinner 3
3 23.68 3.31 Male No Sun Dinner 2
4 24.59 3.61 Female No Sun Dinner 4
  • We can fit a simple linear model to predict tips as a function of total bills:
$$\text{predicted tip} = w_0 + w_1 \cdot \text{total bill}$$
  • This is a reasonable thing to do, because total bills and tips appeared to be linearly associated when we visualized them on a scatter plot a few slides ago.

Recap: Simple linear regression¶

A simple linear regression model is a linear model with a single feature, as we have here. For any total bill $x_i$, the predicted tip $H(x_i)$ is given by

$$H(x_i) = w_0 + w_1x_i$$
  • Question: How do we determine which intercept, $w_0$, and slope, $w_1$, to use?
  • One answer: Pick the $w_0$ and $w_1$ that minimize mean squared error. If $x_i$ and $y_i$ correspond to the $i$th total bill and tip, respectively, then:
$$\begin{align*}\text{MSE} &= \frac{1}{n} \sum_{i = 1}^n \big( y_i - H(x_i) \big)^2 \\ &= \frac{1}{n} \sum_{i = 1}^n \big( y_i - w_0 - w_1x_i \big)^2\end{align*}$$
  • Key idea: The lower the MSE on our training data is, the "better" the model fits the training data.
    • Lower MSE = better predictions.
    • But lower MSE ≠ more reflective of reality!

Empirical risk minimization, by hand¶

$$\begin{align*}\text{MSE} &= \frac{1}{n} \sum_{i = 1}^n \big( y_i - w_0 - w_1x_i \big)^2\end{align*}$$
  • In DSC 40A, you found the formulas for the best intercept, $w_0^*$, and the best slope, $w_1^*$, through calculus.
    • The resulting line, $H(x_i) = w_0^* + w_1^* x_i$, is called the line of best fit, or the regression line.
  • Specifically, if $r$ is the correlation coefficient, $\sigma_x$ and $\sigma_y$ are the standard deviations of $x$ and $y$, and $\bar{x}$ and $\bar{y}$ are the means of $x$ and $y$, then:
$$w_1^* = r \cdot \frac{\sigma_y}{\sigma_x}$$$$w_0^* = \bar{y} - w_1^* \bar{x}$$

Regression in sklearn¶

sklearn¶

No description has been provided for this image
  • sklearn (scikit-learn) implements many common steps in the feature and model creation pipeline.
    • It is widely used throughout industry and academia.
  • It interfaces with numpy arrays, and to an extent, pandas DataFrames.
  • Huge benefit: the documentation online is excellent.

The LinearRegression class¶

  • sklearn comes with several subpackages, including linear_model and tree, each of which contains several classes of models.
  • We'll start with the LinearRegression class from linear_model.
In [13]:
from sklearn.linear_model import LinearRegression
  • Important: From the documentation, we have:

LinearRegression fits a linear model with coefficients w = (w1, …, wp) to minimize the residual sum of squares between the observed targets in the dataset, and the targets predicted by the linear approximation.

In other words, **`LinearRegression` minimizes mean squared error by default**! (Per the documentation, it also includes an intercept term by default.)
In [14]:
LinearRegression?

Fitting a simple linear model¶

First, we must instantiate a LinearRegression object and fit it. By calling fit, we are saying "minimize mean squared error on this dataset and find $w^*$."

In [15]:
model = LinearRegression()

# Note that there are two arguments to fit – X and y!
# (It is not necessary to write X= and y=)
model.fit(X=tips[['total_bill']], y=tips['tip'])
Out[15]:
LinearRegression()

After fitting, we can access $w^*$ – that is, the best slope and intercept.

In [16]:
model.intercept_, model.coef_[0]
Out[16]:
(0.9202696135546731, 0.10502451738435337)

These coefficients tell us that the "best way" (according to squared loss) to make tip predictions using a linear model is using:

$$\text{predicted tip} = 0.92 + 0.105 \cdot \text{total bill}$$

This model predicts that people tip by:

  • Tipping a constant 92 cents.
  • Tipping 10.5% for every dollar spent.

This is the best "linear" pattern in the dataset – it doesn't mean this is actually how people tip!

Let's visualize this model, along with our previous model.

In [17]:
line_pts = pd.DataFrame({'total_bill': [0, 60]})

fig = px.scatter(tips, x='total_bill', y='tip')
fig.add_trace(go.Scatter(
    x=line_pts['total_bill'],
    y=[mean_tip, mean_tip],
    mode='lines',
    name='Constant Model (Mean Tip)'
))
fig.add_trace(go.Scatter(
    x=line_pts['total_bill'],
    y=model.predict(line_pts),
    mode='lines',
    name='Linear Model: Total Bill Only'
))
fig.update_layout(title='Tip vs. Total Bill',
                  xaxis_title='Total Bill', 
                  yaxis_title='Tip')

Visually, our linear model seems to be a better fit for our dataset than our constant model.

Can we quantify whether or not it is better?

Making predictions¶

Fit LinearRegression objects also have a predict method, which can be used to predict tips for any total bill, new or old.

In [18]:
model.predict([[15]])

/Users/sam/mambaforge/envs/dsc80/lib/python3.8/site-packages/sklearn/base.py:450: UserWarning:

X does not have valid feature names, but LinearRegression was fitted with feature names
Out[18]:
array([2.5])
In [19]:
# Since we trained on a DataFrame, the input to model.predict should also
# be a DataFrame. To avoid having to do this, we can use .to_numpy()
# when specifying X= and y=.
test_points = pd.DataFrame({'total_bill': [15, 4, 100]})
model.predict(test_points)
Out[19]:
array([ 2.5 ,  1.34, 11.42])

Comparing models¶

If we want to compute the RMSE of our model on the training data, we need to find its predictions on every row in the training data, tips.

In [20]:
all_preds = model.predict(tips[['total_bill']])
In [21]:
rmse_dict['one feature: total bill'] = rmse(tips['tip'], all_preds)
rmse_dict
Out[21]:
{'constant tip amount': 1.3807999538298958,
 'one feature: total bill': 1.0178504025697377}
  • The RMSE of our simple linear model is lower than that of our constant model, which means it does a better job at predicting tips in our training data than our constant model.
  • Theory tells us it's impossible for the RMSE on the training data to increase as we add more features to the same model. However, the RMSE may increase on unseen data by adding more features; we'll discuss this idea more soon.

💡 Pro-Tip: Making a DF just to display¶

Sometimes it's nice to put your values into a dataframe just to make them display nicely:

In [22]:
rmse_dict
Out[22]:
{'constant tip amount': 1.3807999538298958,
 'one feature: total bill': 1.0178504025697377}
In [23]:
pd.DataFrame({'rmse': rmse_dict.values()}, index=rmse_dict.keys())
Out[23]:
rmse
constant tip amount 1.38
one feature: total bill 1.02

Question 🤔 (Answer at q.dsc80.com)

Remember, you can always ask questions at q.dsc80.com!

Model #3: Multiple linear regression using total bill and table size¶

  • There are still many features in tips we haven't touched:
In [24]:
tips.head()
Out[24]:
total_bill tip sex smoker day time size
0 16.99 1.01 Female No Sun Dinner 2
1 10.34 1.66 Male No Sun Dinner 3
2 21.01 3.50 Male No Sun Dinner 3
3 23.68 3.31 Male No Sun Dinner 2
4 24.59 3.61 Female No Sun Dinner 4
  • Let's try using another feature – table size. Such a model would predict tips using:
$$\text{predicted tip} = w_0 + w_1 \cdot \text{total bill} + w_2 \cdot \text{table size}$$

Multiple linear regression¶

To find the optimal parameters $w^*$, we will again use sklearn's LinearRegression class. The code is not all that different!

In [25]:
model_two = LinearRegression()
model_two.fit(X=tips[['total_bill', 'size']], y=tips['tip'])
Out[25]:
LinearRegression()
In [26]:
model_two.intercept_, model_two.coef_
Out[26]:
(0.6689447408125031, array([0.09, 0.19]))
In [27]:
test_pts = pd.DataFrame({'total_bill': [25], 'size': [4]})
model_two.predict(test_pts)
Out[27]:
array([3.76])

What does this model look like?

Plane of best fit ✈️¶

Here, we must draw a 3D scatter plot and plane, with one axis for total bill, one axis for table size, and one axis for tip. The code below does this.

In [28]:
XX, YY = np.mgrid[0:50:2, 0:8:1]
Z = model_two.intercept_ + model_two.coef_[0] * XX + model_two.coef_[1] * YY
plane = go.Surface(x=XX, y=YY, z=Z, colorscale='Oranges')

fig = go.Figure(data=[plane])
fig.add_trace(go.Scatter3d(x=tips['total_bill'], 
                           y=tips['size'], 
                           z=tips['tip'], mode='markers',
                           marker={'color': '#656DF1', 'size': 5}))

fig.update_layout(scene=dict(xaxis_title='Total Bill',
                             yaxis_title='Table Size',
                             zaxis_title='Tip'),
                  title='Tip vs. Total Bill and Table Size',
                  width=500, height=500)

Comparing models, again¶

How does our two-feature linear model stack up to our single feature linear model and our constant model?

In [29]:
rmse_dict['two features'] = rmse(
    tips['tip'], model_two.predict(tips[['total_bill', 'size']])
)
In [30]:
pd.DataFrame({'rmse': rmse_dict.values()}, index=rmse_dict.keys())
Out[30]:
rmse
constant tip amount 1.38
one feature: total bill 1.02
two features 1.01
  • The RMSE of our two-feature model is the lowest of the three models we've looked at so far, but not by much. We didn't gain much by adding table size to our linear model.
  • It's also not clear whether table sizes are practically useful in predicting tips.
    • We already have the total amount the table spent; why do we need to know how many people were there?

Residual plots¶

  • From DSC 10: one important technique for diagnosing model fit is the residual plot.
  • The $i$th residual is $ y_i - H(x_i) $.
  • A residual plot has
    • predicted values $H(x)$ on the $x$-axis, and
    • residuals $ y - H(x) $ on the $y$-axis.
In [31]:
# Let's start with the single-variable model:
with_resid = tips.assign(**{
    'Predicted Tip': model.predict(tips[['total_bill']]),
    'Residual': tips['tip'] - model.predict(tips[['total_bill']]),
})
fig = px.scatter(with_resid, x='Predicted Tip', y='Residual')
fig.add_hline(0, line_width=2, opacity=1).update_layout(title='Residual Plot (Simple Linear Model)')
  • If all assumptions about linear regression hold, then residual plot should look randomly scattered around the horizontal line $y = 0$.
  • Here, we see that the model makes bigger mistakes for larger predicted values. But overall, there's no apparent trend, so a linear model seems appropriate.
In [32]:
# What about the two-variable model?
with_resid = tips.assign(**{
    'Predicted Tip': model_two.predict(tips[['total_bill', 'size']]),
    'Residual': tips['tip'] - model_two.predict(tips[['total_bill', 'size']]),
})
fig = px.scatter(with_resid, x='Predicted Tip', y='Residual')
fig.add_hline(0, line_width=2, opacity=1).update_layout(title='Residual Plot (Multiple Regression)')

Looks about the same as the previous plot!

Question 🤔 (Answer at q.dsc80.com)

Consider the following dataframe grades that has a midterm and final column.

Ask ChatGPT to make two plots using plotly:

  1. A regular residual plot.
  2. A residual plot that accidentally puts the original final on the x-axis instead of the predicted values.

What do you notice?

In [33]:
grades = pd.read_csv('data/gradesW4315.csv')[['midterm', 'final']]
grades
Out[33]:
midterm final
0 80 103
1 53 79
2 91 122
... ... ...
49 77 98
50 70 134
51 75 99

52 rows × 2 columns

In [ ]:

Summary, next time¶

Summary¶

  • We built three models:
    • A constant model: $\text{predicted tip} = h^*$.
    • A simple linear regression model: $\text{predicted tip} = w_0^* + w_1^* \cdot \text{total bill}$.
    • A multiple linear regression model: $\text{predicted tip} = w_0^* + w_1^* \cdot \text{total bill} + w_2^* \cdot \text{table size}$.
  • As we added more features, our RMSEs decreased.
    • This was guaranteed to happen, since we were only looking at our training data.
  • It is not clear that the final linear model is actually "better"; it doesn't seem to reflect reality better than the previous models.

LinearRegression summary¶

Property Example Description
Initialize model parameters lr = LinearRegression() Create (empty) linear regression model
Fit the model to the data lr.fit(X, y) Determines regression coefficients
Use model for prediction lr.predict(X_new) Uses regression line to make predictions
Evaluate the model lr.score(X, y) Calculates the $R^2$ of the LR model
Access model attributes lr.coef_, lr.intercept_ Accesses the regression coefficients and intercept

Next time¶

In [34]:
tips.head()
Out[34]:
total_bill tip sex smoker day time size
0 16.99 1.01 Female No Sun Dinner 2
1 10.34 1.66 Male No Sun Dinner 3
2 21.01 3.50 Male No Sun Dinner 3
3 23.68 3.31 Male No Sun Dinner 2
4 24.59 3.61 Female No Sun Dinner 4

  • So far, in our journey to predict 'tip', we've only used the existing numerical features in our dataset, 'total_bill' and 'size'.

  • There's a lot of information in tips that we didn't use – 'sex', 'smoker', 'day', and 'time', for example. We can't use these features in their current form, because they're non-numeric.

  • How do we use categorical features in a regression model?