Generalized Additive Models#

This module has helper functions to help compute the degrees of freedom of a GAM and to create a partial dependence plot of a fitted pygam model.

import numpy as np
from pygam import LinearGAM, s
from ISLP.pygam import (plot, 
                        approx_lam, 
                        degrees_of_freedom)

Make a toy dataset#

We create a simple dataset with 5 features. We’ll have a cubic effect for our first feature, and linear for the remaining 4 features.

By construction, all the “action” in our GAM will be in the first feature. This will have our scatter plot look like the partial residuals from our fit. Usually, the scatter plot will not look so nice on a partial dependence plot. One should use partial residuals instead. We take this liberty here while demonstrating the plot function.

rng = np.random.default_rng(1)
N = 100
X = rng.normal(size=(N, 3))
Y = X[:,0] + 0.3 * X[:,0]**3 + rng.normal(size=N)

Create a GAM#

Let’s start of fitting a GAM with a relatively small amount of smoothing.

terms = [s(f, lam=0.01) for f in range(3)]
gam = LinearGAM(terms[0] + 
                terms[1] + 
                terms[2])
gam.fit(X, Y)

LinearGAM(callbacks=[Deviance(), Diffs()], fit_intercept=True, 
   max_iter=100, scale=None, terms=s(0) + s(1) + s(2) + intercept, 
   tol=0.0001, verbose=False)

Plot the partial dependence plot for first feature#

ax = plot(gam, 0)

../_images/f6b63453113945b8dbf6a45c9286bd222ae293317813d539b56a65e8b42c3812.png

Including a scatter plot of

ax.scatter(X[:,0], 
           Y - Y.mean(),
          facecolor='k',
          alpha=0.4)
ax.get_figure()

../_images/bd7395635b6a3dcad2c15047137c658a4313ef4897f9e83c24c8c45f409b87b7.png

Let’s take a look at (approximately) how many degrees of freedom we’ve used:

[degrees_of_freedom(X,
                   terms[i]) for i in range(X.shape[1])]

[14.217770998911778, 14.464699512943346, 16.489388425772507]

Fixing degrees of freedom#

Suppose we want to use 5 degrees of freedom for each feature. We compute a value of lam for each that fixes the degrees of freedom at 5.

lam_vals = [approx_lam(X,
                       terms[i],
                       df=5) for i in range(X.shape[1])]
lam_vals

[array([63.50400093]), array([26.02830335]), array([22.44251121])]

Create a new GAM with the correctly fixed terms#

fixed_terms = [s(f, lam=l) for 
               f, l in zip(range(3), lam_vals)]
fixed_gam = LinearGAM(fixed_terms[0] + 
                      fixed_terms[1] + 
                      fixed_terms[2])
fixed_gam.fit(X, Y)

LinearGAM(callbacks=[Deviance(), Diffs()], fit_intercept=True, 
   max_iter=100, scale=None, terms=s(0) + s(1) + s(2) + intercept, 
   tol=0.0001, verbose=False)

ax = plot(fixed_gam, 0)
ax.scatter(X[:,0], 
           Y - Y.mean(),
          facecolor='k',
          alpha=0.4);

../_images/9321d7d1616d97dec688e15b6f26bd63c5a47ed8478a7e44bebd93d92e2cc21e.png

Generalized Additive Models

Contents