#!/usr/bin/env python
# coding: utf-8

# # Regularization

# ## Agenda:
# 
# 1. Overfitting (review)
# 2. Overfitting with linear models
# 3. Regularization of linear models
# 4. Regularized regression in scikit-learn
# 5. Regularized classification in scikit-learn
# 6. Comparing regularized linear models with unregularized linear models

# ## Part 1: Overfitting (review)
# 
# **What is overfitting?**
# 
# - Building a model that matches the training data "too closely"
# - Learning from the noise in the data, rather than just the signal
# 
# **How does overfitting occur?**
# 
# - Evaluating a model by testing it on the same data that was used to train it
# - Creating a model that is "too complex"
# 
# **What is the impact of overfitting?**
# 
# - Model will do well on the training data, but won't generalize to out-of-sample data
# - Model will have low bias, but high variance

# ### Overfitting with KNN
# 
# ![Overfitting with KNN](images/iris_01nn_map.png)

# ### Overfitting with polynomial regression
# 
# ![Overfitting with polynomial regression](images/polynomial_overfitting.png)

# ### Overfitting with decision trees
# 
# ![Overfitting with decision trees](images/salary_tree_deep.png)

# ## Part 2: Overfitting with linear models
# 
# **What are the general characteristics of linear models?**
# 
# - Low model complexity
# - High bias, low variance
# - Does not tend to overfit
# 
# Nevertheless, **overfitting can still occur** with linear models if you allow them to have **high variance**. Here are some common causes:

# ### Cause 1: Irrelevant features
# 
# Linear models can overfit if you include "irrelevant features", meaning features that are unrelated to the response. Why?
# 
# Because it will learn a coefficient for every feature you include in the model, regardless of whether that feature has the **signal** or the **noise**.
# 
# This is especially a problem when **p (number of features) is close to n (number of observations)**, because that model will naturally have high variance.

# ### Cause 2: Correlated features
# 
# Linear models can overfit if the included features are highly correlated with one another. Why?
# 
# From the [scikit-learn documentation](http://scikit-learn.org/stable/modules/linear_model.html#ordinary-least-squares):
# 
# > "...coefficient estimates for Ordinary Least Squares rely on the independence of the model terms. When terms are correlated and the columns of the design matrix X have an approximate linear dependence, the design matrix becomes close to singular and as a result, the least-squares estimate becomes highly sensitive to random errors in the observed response, producing a large variance."

# ### Cause 3: Large coefficients
# 
# Linear models can overfit if the coefficients (after feature standardization) are too large. Why?
# 
# Because the **larger** the absolute value of the coefficient, the more **power** it has to change the predicted response, resulting in a higher variance.

# ## Part 3: Regularization of linear models
# 
# - Regularization is a method for "constraining" or "regularizing" the **size of the coefficients**, thus "shrinking" them towards zero.
# - It reduces model variance and thus **minimizes overfitting**.
# - If the model is too complex, it tends to reduce variance more than it increases bias, resulting in a model that is **more likely to generalize**.
# 
# Our goal is to locate the **optimum model complexity**, and thus regularization is useful when we believe our model is too complex.

# ![Bias-variance tradeoff](images/bias_variance.png)

# ### How does regularization work?
# 
# For a normal linear regression model, we estimate the coefficients using the least squares criterion, which **minimizes the residual sum of squares (RSS):**

# ![Estimating coefficients](images/estimating_coefficients.png)

# For a regularized linear regression model, we **minimize the sum of RSS and a "penalty term"** that penalizes coefficient size.
# 
# **Ridge regression** (or "L2 regularization") minimizes: $$\text{RSS} + \alpha \sum_{j=1}^p \beta_j^2$$
# 
# **Lasso regression** (or "L1 regularization") minimizes: $$\text{RSS} + \alpha \sum_{j=1}^p |\beta_j|$$
# 
# - $p$ is the **number of features**
# - $\beta_j$ is a **model coefficient**
# - $\alpha$ is a **tuning parameter:**
#     - A tiny $\alpha$ imposes no penalty on the coefficient size, and is equivalent to a normal linear regression model.
#     - Increasing the $\alpha$ penalizes the coefficients and thus shrinks them.

# ### Lasso and ridge path diagrams
# 
# A larger alpha (towards the left of each diagram) results in more regularization:
# 
# - **Lasso regression** shrinks coefficients all the way to zero, thus removing them from the model
# - **Ridge regression** shrinks coefficients toward zero, but they rarely reach zero
# 
# Source code for the diagrams: [Lasso regression](http://scikit-learn.org/stable/auto_examples/linear_model/plot_lasso_lars.html) and [Ridge regression](http://scikit-learn.org/stable/auto_examples/linear_model/plot_ridge_path.html)

# ![Lasso and Ridge Path Diagrams](images/lasso_ridge_path.png)

# ### Advice for applying regularization
# 
# **Should features be standardized?**
# 
# - Yes, because otherwise, features would be penalized simply because of their scale.
# - Also, standardizing avoids penalizing the intercept, which wouldn't make intuitive sense.
# 
# **How should you choose between Lasso regression and Ridge regression?**
# 
# - Lasso regression is preferred if we believe many features are irrelevant or if we prefer a sparse model.
# - If model performance is your primary concern, it is best to try both.
# - ElasticNet regression is a combination of lasso regression and ridge Regression.

# ### Visualizing regularization
# 
# Below is a visualization of what happens when you apply regularization. The general idea is that you are **restricting the allowed values of your coefficients** to a certain "region". **Within that region**, you want to find the coefficients that result in the best model.

# ![Lasso and Ridge Coefficient Plots](images/lasso_ridge_coefficients.png)

# In this diagram:
# 
# - We are fitting a linear regression model with **two features**, $x_1$ and $x_2$.
# - $\hat\beta$ represents the set of two coefficients, $\beta_1$ and $\beta_2$, which minimize the RSS for the **unregularized model**.
# - Regularization restricts the allowed positions of $\hat\beta$ to the **blue constraint region:**
#     - For lasso, this region is a **diamond** because it constrains the absolute value of the coefficients.
#     - For ridge, this region is a **circle** because it constrains the square of the coefficients.
# - The **size of the blue region** is determined by $\alpha$, with a smaller $\alpha$ resulting in a larger region:
#     - When $\alpha$ is zero, the blue region is infinitely large, and thus the coefficient sizes are not constrained.
#     - When $\alpha$ increases, the blue region gets smaller and smaller.
# 
# In this case, $\hat\beta$ is **not** within the blue constraint region. Thus, we need to **move $\hat\beta$ until it intersects the blue region**, while **increasing the RSS as little as possible.**
# 
# From page 222 of [An Introduction to Statistical Learning](http://www-bcf.usc.edu/~gareth/ISL/):
# 
# > The ellipses that are centered around $\hat\beta$ represent **regions of constant RSS**. In other words, all of the points on a given ellipse share a common value of the RSS. As the ellipses expand away from the least squares coefficient estimates, the RSS increases. Equations (6.8) and (6.9) indicate that the lasso and ridge regression coefficient estimates are given by the **first point at which an ellipse contacts the constraint region**.
# 
# > Since **ridge regression** has a circular constraint with no sharp points, this intersection will not generally occur on an axis, and so the ridge regression coefficient estimates will be exclusively non-zero. However, the **lasso** constraint has corners at each of the axes, and so the ellipse will often intersect the constraint region at an axis. When this occurs, one of the coefficients will equal zero. In higher dimensions, many of the coefficient estimates may equal zero simultaneously. In Figure 6.7, the intersection occurs at $\beta_1 = 0$, and so the resulting model will only include $\beta_2$.

# ## Part 4: Regularized regression in scikit-learn

# - Communities and Crime dataset from the UCI Machine Learning Repository: [data](http://archive.ics.uci.edu/ml/machine-learning-databases/communities/communities.data), [data dictionary](http://archive.ics.uci.edu/ml/datasets/Communities+and+Crime)
# - **Goal:** Predict the violent crime rate for a community given socioeconomic and law enforcement data

# ### Load and prepare the crime dataset

# In[1]:


# read in the dataset
import pandas as pd
url = 'http://archive.ics.uci.edu/ml/machine-learning-databases/communities/communities.data'
crime = pd.read_csv(url, header=None, na_values=['?'])
crime.head()


# In[2]:


# examine the response variable
crime[127].describe()


# In[3]:


# remove categorical features
crime.drop([0, 1, 2, 3, 4], axis=1, inplace=True)


# In[4]:


# remove rows with any missing values
crime.dropna(inplace=True)


# In[5]:


# check the shape
crime.shape


# In[6]:


# define X and y
X = crime.drop(127, axis=1)
y = crime[127]


# In[7]:


# split into training and testing sets
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=1)


# ### Linear regression

# In[8]:


# build a linear regression model
from sklearn.linear_model import LinearRegression
linreg = LinearRegression()
linreg.fit(X_train, y_train)


# In[9]:


# examine the coefficients
print linreg.coef_


# In[10]:


# make predictions
y_pred = linreg.predict(X_test)


# In[11]:


# calculate RMSE
from sklearn import metrics
import numpy as np
print np.sqrt(metrics.mean_squared_error(y_test, y_pred))


# ### Ridge regression
# 
# - [Ridge](http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.Ridge.html) documentation
# - **alpha:** must be positive, increase for more regularization
# - **normalize:** scales the features (without using StandardScaler)

# In[12]:


# alpha=0 is equivalent to linear regression
from sklearn.linear_model import Ridge
ridgereg = Ridge(alpha=0, normalize=True)
ridgereg.fit(X_train, y_train)
y_pred = ridgereg.predict(X_test)
print np.sqrt(metrics.mean_squared_error(y_test, y_pred))


# In[13]:


# try alpha=0.1
ridgereg = Ridge(alpha=0.1, normalize=True)
ridgereg.fit(X_train, y_train)
y_pred = ridgereg.predict(X_test)
print np.sqrt(metrics.mean_squared_error(y_test, y_pred))


# In[14]:


# examine the coefficients
print ridgereg.coef_


# - [RidgeCV](http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.RidgeCV.html): ridge regression with built-in cross-validation of the alpha parameter
# - **alphas:** array of alpha values to try

# In[15]:


# create an array of alpha values
alpha_range = 10.**np.arange(-2, 3)
alpha_range


# In[16]:


# select the best alpha with RidgeCV
from sklearn.linear_model import RidgeCV
ridgeregcv = RidgeCV(alphas=alpha_range, normalize=True, scoring='mean_squared_error')
ridgeregcv.fit(X_train, y_train)
ridgeregcv.alpha_


# In[17]:


# predict method uses the best alpha value
y_pred = ridgeregcv.predict(X_test)
print np.sqrt(metrics.mean_squared_error(y_test, y_pred))


# ### Lasso regression
# 
# - [Lasso](http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.Lasso.html) documentation
# - **alpha:** must be positive, increase for more regularization
# - **normalize:** scales the features (without using StandardScaler)

# In[18]:


# try alpha=0.001 and examine coefficients
from sklearn.linear_model import Lasso
lassoreg = Lasso(alpha=0.001, normalize=True)
lassoreg.fit(X_train, y_train)
print lassoreg.coef_


# In[19]:


# try alpha=0.01 and examine coefficients
lassoreg = Lasso(alpha=0.01, normalize=True)
lassoreg.fit(X_train, y_train)
print lassoreg.coef_


# In[20]:


# calculate RMSE (for alpha=0.01)
y_pred = lassoreg.predict(X_test)
print np.sqrt(metrics.mean_squared_error(y_test, y_pred))


# - [LassoCV](http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LassoCV.html): lasso regression with built-in cross-validation of the alpha parameter
# - **n_alphas:** number of alpha values (automatically chosen) to try

# In[21]:


# select the best alpha with LassoCV
from sklearn.linear_model import LassoCV
lassoregcv = LassoCV(n_alphas=100, normalize=True, random_state=1)
lassoregcv.fit(X_train, y_train)
lassoregcv.alpha_


# In[22]:


# examine the coefficients
print lassoregcv.coef_


# In[23]:


# predict method uses the best alpha value
y_pred = lassoregcv.predict(X_test)
print np.sqrt(metrics.mean_squared_error(y_test, y_pred))


# ## Part 5: Regularized classification in scikit-learn
# 
# - Wine dataset from the UCI Machine Learning Repository: [data](http://archive.ics.uci.edu/ml/machine-learning-databases/wine/wine.data), [data dictionary](http://archive.ics.uci.edu/ml/datasets/Wine)
# - **Goal:** Predict the origin of wine using chemical analysis

# ### Load and prepare the wine dataset

# In[24]:


# read in the dataset
url = 'http://archive.ics.uci.edu/ml/machine-learning-databases/wine/wine.data'
wine = pd.read_csv(url, header=None)
wine.head()


# In[25]:


# examine the response variable
wine[0].value_counts()


# In[26]:


# define X and y
X = wine.drop(0, axis=1)
y = wine[0]


# In[27]:


# split into training and testing sets
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y, random_state=1)


# ### Logistic regression (unregularized)

# In[28]:


# build a logistic regression model
from sklearn.linear_model import LogisticRegression
logreg = LogisticRegression(C=1e9)
logreg.fit(X_train, y_train)


# In[29]:


# examine the coefficients
print logreg.coef_


# In[30]:


# generate predicted probabilities
y_pred_prob = logreg.predict_proba(X_test)
print y_pred_prob


# In[31]:


# calculate log loss
print metrics.log_loss(y_test, y_pred_prob)


# ### Logistic regression (regularized)
# 
# - [LogisticRegression](http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) documentation
# - **C:** must be positive, decrease for more regularization
# - **penalty:** l1 (lasso) or l2 (ridge)

# In[32]:


# standardize X_train and X_test
from sklearn.preprocessing import StandardScaler
scaler = StandardScaler()
scaler.fit(X_train)
X_train_scaled = scaler.transform(X_train)
X_test_scaled = scaler.transform(X_test)


# In[33]:


# try C=0.1 with L1 penalty
logreg = LogisticRegression(C=0.1, penalty='l1')
logreg.fit(X_train_scaled, y_train)
print logreg.coef_


# In[34]:


# generate predicted probabilities and calculate log loss
y_pred_prob = logreg.predict_proba(X_test_scaled)
print metrics.log_loss(y_test, y_pred_prob)


# In[35]:


# try C=0.1 with L2 penalty
logreg = LogisticRegression(C=0.1, penalty='l2')
logreg.fit(X_train_scaled, y_train)
print logreg.coef_


# In[36]:


# generate predicted probabilities and calculate log loss
y_pred_prob = logreg.predict_proba(X_test_scaled)
print metrics.log_loss(y_test, y_pred_prob)


# - [Pipeline](http://scikit-learn.org/stable/modules/pipeline.html): chain steps together
# - [GridSearchCV](http://scikit-learn.org/stable/modules/grid_search.html): search a grid of parameters

# In[37]:


# pipeline of StandardScaler and LogisticRegression
from sklearn.pipeline import make_pipeline
pipe = make_pipeline(StandardScaler(), LogisticRegression())


# In[38]:


# grid search for best combination of C and penalty
from sklearn.grid_search import GridSearchCV
C_range = 10.**np.arange(-2, 3)
penalty_options = ['l1', 'l2']
param_grid = dict(logisticregression__C=C_range, logisticregression__penalty=penalty_options)
grid = GridSearchCV(pipe, param_grid, cv=10, scoring='log_loss')
grid.fit(X, y)


# In[39]:


# print all log loss scores
grid.grid_scores_


# In[40]:


# examine the best model
print grid.best_score_
print grid.best_params_


# ## Part 6: Comparing regularized linear models with unregularized linear models
# 
# **Advantages of regularized linear models:**
# 
# - Better performance
# - L1 regularization performs automatic feature selection
# - Useful for high-dimensional problems (p > n)
# 
# **Disadvantages of regularized linear models:**
# 
# - Tuning is required
# - Feature scaling is recommended
# - Less interpretable (due to feature scaling)