In [0]:
%matplotlib inline
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import warnings
warnings.simplefilter('ignore', DeprecationWarning)
Let us have a look at the Titanic dataset from the Kaggle Getting Started challenge at:
https://www.kaggle.com/c/titanic-gettingStarted
We can load the CSV file as a pandas data frame in one line:
In [1]:
#!curl -s https://dl.dropboxusercontent.com/u/5743203/data/titanic/titanic_train.csv | head -5
with open('titanic_train.csv', 'r') as f:
for i, line in zip(range(5), f):
print(line.strip())
In [2]:
#data = pd.read_csv('https://dl.dropboxusercontent.com/u/5743203/data/titanic/titanic_train.csv')
data = pd.read_csv('titanic_train.csv')
pandas data frames have a HTML table representation in the IPython notebook. Let's have a look at the first 5 rows:
In [3]:
data.head(5)
Out[3]:
In [4]:
data.count()
Out[4]:
The data frame has 891 rows. Some passengers have missing information though: in particular Age and Cabin info can be missing. The meaning of the columns is explained on the challenge website:
https://www.kaggle.com/c/titanic-gettingStarted/data
A data frame can be converted into a numpy array by calling the values attribute:
In [5]:
list(data.columns)
Out[5]:
In [6]:
data.shape
Out[6]:
In [7]:
data.values
Out[7]:
However this cannot be directly fed to a scikit-learn model:
the target variable (survival) is mixed with the input data
some attribute such as unique ids have no predictive values for the task
the values are heterogeneous (string labels for categories, integers and floating point numbers)
some attribute values are missing (nan: "not a number")
The goal of the challenge is to predict whether a passenger has survived from others known attribute. Let us have a look at the Survived columns:
In [8]:
survived_column = data['Survived']
survived_column.dtype
Out[8]:
data.Survived is an instance of the pandas Series class with an integer dtype:
In [9]:
type(survived_column)
Out[9]:
The data object is an instance pandas DataFrame class:
In [10]:
type(data)
Out[10]:
Series can be seen as homegeneous, 1D columns. DataFrame instances are heterogenous collections of columns with the same length.
The original data frame can be aggregated by counting rows for each possible value of the Survived column:
In [11]:
data.groupby('Survived').count()
Out[11]:
In [12]:
np.mean(survived_column == 0)
Out[12]:
From this the subset of the full passengers list, about 2/3 perished in the event. So if we are to build a predictive model from this data, a baseline model to compare the performance to would be to always predict death. Such a constant model would reach around 62% predictive accuracy (which is higher than predicting at random):
pandas Series instances can be converted to regular 1D numpy arrays by using the values attribute:
In [13]:
target = survived_column.values
In [14]:
type(target)
Out[14]:
In [15]:
target.dtype
Out[15]:
In [16]:
target[:5]
Out[16]:
sklearn estimators all work with homegeneous numerical feature descriptors passed as a numpy array. Therefore passing the raw data frame will not work out of the box.
Let us start simple and build a first model that only uses readily available numerical features as input, namely data['Fare'], data['Pclass'] and data['Age'].
In [17]:
numerical_features = data[['Fare', 'Pclass', 'Age']]
numerical_features.head(5)
Out[17]:
Unfortunately some passengers do not have age information:
In [18]:
numerical_features.count()
Out[18]:
Let's use pandas fillna method to input the median age for those passengers:
In [19]:
median_features = numerical_features.dropna().median()
median_features
Out[19]:
In [20]:
imputed_features = numerical_features.fillna(median_features)
imputed_features.count()
Out[20]:
In [21]:
imputed_features.head(5)
Out[21]:
Now that the data frame is clean, we can convert it into an homogeneous numpy array of floating point values:
In [22]:
features_array = imputed_features.values
features_array
Out[22]:
In [23]:
features_array.dtype
Out[23]:
Let's take the 80% of the data for training a first model and keep 20% for computing is generalization score:
In [24]:
from sklearn.cross_validation import train_test_split
features_train, features_test, target_train, target_test = train_test_split(
features_array, target, test_size=0.20, random_state=0)
In [25]:
features_train.shape
Out[25]:
In [26]:
features_test.shape
Out[26]:
In [27]:
target_train.shape
Out[27]:
In [28]:
target_test.shape
Out[28]:
Let's start with a simple model from sklearn, namely LogisticRegression:
In [29]:
from sklearn.linear_model import LogisticRegression
logreg = LogisticRegression(C=1)
logreg.fit(features_train, target_train)
Out[29]:
In [30]:
target_predicted = logreg.predict(features_test)
In [31]:
from sklearn.metrics import accuracy_score
accuracy_score(target_test, target_predicted)
Out[31]:
This first model has around 73% accuracy: this is better than our baseline that always predicts death.
In [32]:
logreg.score(features_test, target_test)
Out[32]:
The coef_ attribute of a fitted linear model such as LogisticRegression holds the weights of each features:
In [33]:
feature_names = numerical_features.columns
feature_names
Out[33]:
In [34]:
logreg.coef_
Out[34]:
In [35]:
x = np.arange(len(feature_names))
plt.bar(x, logreg.coef_.ravel())
plt.xticks(x + 0.5, feature_names, rotation=30);
In this case, survival is slightly positively linked with Fare (the higher the fare, the higher the likelyhood the model will predict survival) while passenger from first class and lower ages are predicted to survive more often than older people from the 3rd class.
First-class cabins were closer to the lifeboats and children and women reportedly had the priority. Our model seems to capture that historical data. We will see later if the sex of the passenger can be used as an informative predictor to increase the predictive accuracy of the model.
It is possible to see the details of the false positive and false negative errors by computing the confusion matrix:
In [36]:
from sklearn.metrics import confusion_matrix
cm = confusion_matrix(target_test, target_predicted)
print(cm)
The true labeling are seen as the rows and the predicted labels are the columns:
In [37]:
def plot_confusion(cm, target_names = ['survived', 'not survived'],
title='Confusion matrix'):
plt.imshow(cm, interpolation='nearest', cmap=plt.cm.Blues)
plt.title(title)
plt.colorbar()
tick_marks = np.arange(len(target_names))
plt.xticks(tick_marks, target_names, rotation=60)
plt.yticks(tick_marks, target_names)
plt.ylabel('True label')
plt.xlabel('Predicted label')
# Convenience function to adjust plot parameters for a clear layout.
plt.tight_layout()
plot_confusion(cm)
In [38]:
print(cm)
We can normalize the number of prediction by dividing by the total number of true "survived" and "not survived" to compute true and false positive rates for survival in the first row and the false negative and true negative rates in the second row.
In [39]:
cm.sum(axis=1)
Out[39]:
In [40]:
cm_normalized = cm.astype(np.float64) / cm.sum(axis=1)[:, np.newaxis]
print(cm_normalized)
In [41]:
plot_confusion(cm_normalized, title="Normalized confusion matrix")
We can therefore observe that the fact that the target classes are not balanced in the dataset makes the accuracy score not very informative.
scikit-learn provides alternative classification metrics to evaluate models performance on imbalanced data such as precision, recall and f1 score:
In [42]:
from sklearn.metrics import classification_report
print(classification_report(target_test, target_predicted,
target_names=['not survived', 'survived']))
Another way to quantify the quality of a binary classifier on imbalanced data is to compute the precision, recall and f1-score of a model (at the default fixed decision threshold of 0.5).
Logistic Regression is a probabilistic models: instead of just predicting a binary outcome (survived or not) given the input features it can also estimates the posterior probability of the outcome given the input features using the predict_proba method:
In [43]:
target_predicted_proba = logreg.predict_proba(features_test)
target_predicted_proba[:5]
Out[43]:
By default the decision threshold is 0.5: if we vary the decision threshold from 0 to 1 we could generate a family of binary classifier models that address all the possible trade offs between false positive and false negative prediction errors.
We can summarize the performance of a binary classifier for all the possible thresholds by plotting the ROC curve and quantifying the Area under the ROC curve:
In [44]:
from sklearn.metrics import roc_curve
from sklearn.metrics import auc
def plot_roc_curve(target_test, target_predicted_proba):
fpr, tpr, thresholds = roc_curve(target_test, target_predicted_proba[:, 1])
roc_auc = auc(fpr, tpr)
# Plot ROC curve
plt.plot(fpr, tpr, label='ROC curve (area = %0.3f)' % roc_auc)
plt.plot([0, 1], [0, 1], 'k--') # random predictions curve
plt.xlim([0.0, 1.0])
plt.ylim([0.0, 1.0])
plt.xlabel('False Positive Rate or (1 - Specifity)')
plt.ylabel('True Positive Rate or (Sensitivity)')
plt.title('Receiver Operating Characteristic')
plt.legend(loc="lower right")
In [45]:
plot_roc_curve(target_test, target_predicted_proba)
Here the area under ROC curve is 0.756 which is very similar to the accuracy (0.732). However the ROC-AUC score of a random model is expected to 0.5 on average while the accuracy score of a random model depends on the class imbalance of the data. ROC-AUC can be seen as a way to callibrate the predictive accuracy of a model against class imbalance.
We previously decided to randomly split the data to evaluate the model on 20% of held-out data. However the location randomness of the split might have a significant impact in the estimated accuracy:
In [46]:
features_train, features_test, target_train, target_test = train_test_split(
features_array, target, test_size=0.20, random_state=0)
logreg.fit(features_train, target_train).score(features_test, target_test)
Out[46]:
In [47]:
features_train, features_test, target_train, target_test = train_test_split(
features_array, target, test_size=0.20, random_state=1)
logreg.fit(features_train, target_train).score(features_test, target_test)
Out[47]:
In [48]:
features_train, features_test, target_train, target_test = train_test_split(
features_array, target, test_size=0.20, random_state=2)
logreg.fit(features_train, target_train).score(features_test, target_test)
Out[48]:
So instead of using a single train / test split, we can use a group of them and compute the min, max and mean scores as an estimation of the real test score while not underestimating the variability:
In [49]:
from sklearn.cross_validation import cross_val_score
scores = cross_val_score(logreg, features_array, target, cv=5)
scores
Out[49]:
In [50]:
scores.min(), scores.mean(), scores.max()
Out[50]:
cross_val_score reports accuracy by default be it can also be used to report other performance metrics such as ROC-AUC or f1-score:
In [51]:
scores = cross_val_score(logreg, features_array, target, cv=5,
scoring='roc_auc')
scores.min(), scores.mean(), scores.max()
Out[51]:
Exercise:
Compute cross-validated scores for other classification metrics ('precision', 'recall', 'f1', 'accuracy'...).
Change the number of cross-validation folds between 3 and 10: what is the impact on the mean score? on the processing time?
Hints:
The list of classification metrics is available in the online documentation:
http://scikit-learn.org/stable/modules/model_evaluation.html#common-cases-predefined-values
You can use the %%time cell magic on the first line of an IPython cell to measure the time of the execution of the cell.
In [52]:
Let us now try to build richer models by including more features as potential predictors for our model.
Categorical variables such as data['Embarked'] or data['Sex'] can be converted as boolean indicators features also known as dummy variables or one-hot-encoded features:
In [53]:
pd.get_dummies(data['Sex'], prefix='Sex').head(5)
Out[53]:
In [54]:
pd.get_dummies(data.Embarked, prefix='Embarked').head(5)
Out[54]:
We can combine those new numerical features with the previous features using pandas.concat along axis=1:
In [55]:
rich_features = pd.concat([data[['Fare', 'Pclass', 'Age']],
pd.get_dummies(data['Sex'], prefix='Sex'),
pd.get_dummies(data['Embarked'], prefix='Embarked')],
axis=1)
rich_features.head(5)
Out[55]:
By construction the new Sex_male feature is redundant with Sex_female. Let us drop it:
In [56]:
rich_features_no_male = rich_features.drop('Sex_male', 1)
rich_features_no_male.head(5)
Out[56]:
Let us not forget to imput the median age for passengers without age information:
In [57]:
rich_features_final = rich_features_no_male.fillna(rich_features_no_male.dropna().median())
rich_features_final.head(5)
Out[57]:
We can finally cross-validate a logistic regression model on this new data an observe that the mean score has significantly increased:
In [58]:
%%time
from sklearn.linear_model import LogisticRegression
from sklearn.cross_validation import cross_val_score
logreg = LogisticRegression(C=1)
scores = cross_val_score(logreg, rich_features_final, target, cv=5, scoring='accuracy')
print("Logistic Regression CV scores:")
print("min: {:.3f}, mean: {:.3f}, max: {:.3f}".format(
scores.min(), scores.mean(), scores.max()))
Exercise:
change the value of the parameter C. Does it have an impact on the score?
fit a new instance of the logistic regression model on the full dataset.
plot the weights for the features of this newly fitted logistic regression model.
In [59]:
%load solutions/04A_plot_logistic_regression_weights.py
In [60]:
sklearn also implement non linear models that are known to perform very well for data-science projects where datasets have not too many features (e.g. less than 5000).
In particular let us have a look at Random Forests and Gradient Boosted Trees:
In [61]:
%%time
from sklearn.ensemble import RandomForestClassifier
rf = RandomForestClassifier(n_estimators=100)
scores = cross_val_score(rf, rich_features_final, target, cv=5, n_jobs=4,
scoring='accuracy')
print("Random Forest CV scores:")
print("min: {:.3f}, mean: {:.3f}, max: {:.3f}".format(
scores.min(), scores.mean(), scores.max()))
In [62]:
%%time
from sklearn.ensemble import GradientBoostingClassifier
gb = GradientBoostingClassifier(n_estimators=100, learning_rate=0.1,
subsample=.8, max_features=.5)
scores = cross_val_score(gb, rich_features_final, target, cv=5, n_jobs=4,
scoring='accuracy')
print("Gradient Boosted Trees CV scores:")
print("min: {:.3f}, mean: {:.3f}, max: {:.3f}".format(
scores.min(), scores.mean(), scores.max()))
Both models seem to do slightly better than the logistic regression model on this data.
Exercise:
Change the value of the learning_rate and other GradientBoostingClassifier parameter, can you get a better mean score?
Would treating the PClass variable as categorical improve the models performance?
Find out which predictor variables (features) are the most informative for those models.
Hints:
Fitted ensembles of trees have feature_importances_ attribute that can be used similarly to the coef_ attribute of linear models.
In [63]:
%load solutions/04B_more_categorical_variables.py
In [64]:
In [65]:
%load solutions/04C_feature_importance.py
In [66]:
Instead of changing the value of the learning rate manually and re-running the cross-validation, we can find the best values for the parameters automatically (assuming we are ready to wait):
In [67]:
%%time
from sklearn.grid_search import GridSearchCV
gb = GradientBoostingClassifier(n_estimators=100, subsample=.8)
params = {
'learning_rate': [0.05, 0.1, 0.5],
'max_features': [0.5, 1],
'max_depth': [3, 4, 5],
}
gs = GridSearchCV(gb, params, cv=5, scoring='roc_auc', n_jobs=4)
gs.fit(rich_features_final, target)
Let us sort the models by mean validation score:
In [68]:
sorted(gs.grid_scores_, key=lambda x: x.mean_validation_score, reverse=True)
Out[68]:
In [69]:
gs.best_score_
Out[69]:
In [70]:
gs.best_params_
Out[70]:
We should note that the mean scores are very close to one another and almost always within one standard deviation of one another. This means that all those parameters are quite reasonable. The only parameter of importance seems to be the learning_rate: 0.5 seems to be a bit too high.
When doing imputation in pandas, prior to computing the train test split we use data from the test to improve the accuracy of the median value that we impute on the training set. This is actually cheating. To avoid this we should compute the median of the features on the training fold and use that median value to do the imputation both on the training and validation fold for a given CV split.
To do this we can prepare the features as previously but without the imputation: we just replace missing values by the -1 marker value:
In [71]:
features = pd.concat([data[['Fare', 'Age']],
pd.get_dummies(data['Sex'], prefix='Sex'),
pd.get_dummies(data['Pclass'], prefix='Pclass'),
pd.get_dummies(data['Embarked'], prefix='Embarked')],
axis=1)
features = features.drop('Sex_male', 1)
# Because of the following bug we cannot use NaN as the missing
# value marker, use a negative value as marker instead:
# https://github.com/scikit-learn/scikit-learn/issues/3044
features = features.fillna(-1)
features.head(5)
Out[71]:
We can now use the Imputer transformer of scikit-learn to find the median value on the training set and apply it on missing values of both the training set and the test set.
In [72]:
from sklearn.cross_validation import train_test_split
X_train, X_test, y_train, y_test = train_test_split(features.values, target, random_state=0)
In [73]:
from sklearn.preprocessing import Imputer
imputer = Imputer(strategy='median', missing_values=-1)
imputer.fit(X_train)
Out[73]:
The median age computed on the training set is stored in the statistics_ attribute.
In [74]:
imputer.statistics_
Out[74]:
In [75]:
features.columns.values
Out[75]:
Imputation can now happen by calling the transform method:
In [76]:
X_train_imputed = imputer.transform(X_train)
X_test_imputed = imputer.transform(X_test)
In [77]:
np.any(X_train == -1)
Out[77]:
In [78]:
np.any(X_train_imputed == -1)
Out[78]:
In [79]:
np.any(X_test == -1)
Out[79]:
In [80]:
np.any(X_test_imputed == -1)
Out[80]:
We can now use a pipeline that wraps an imputer transformer and the classifier itself:
In [81]:
from sklearn.pipeline import Pipeline
imputer = Imputer(strategy='median', missing_values=-1)
classifier = GradientBoostingClassifier(n_estimators=100, learning_rate=0.1,
subsample=.8, max_features=.5,
random_state=0)
pipeline = Pipeline([
('imp', imputer),
('clf', classifier),
])
scores = cross_val_score(pipeline, features.values, target, cv=5, n_jobs=4,
scoring='accuracy', )
print(scores.min(), scores.mean(), scores.max())
The mean cross-validation is slightly lower than we used the imputation on the whole data as we did earlier although not by much. This means that in this case the data-snooping was not really helping the model cheat by much.
Let us re-run the grid search, this time on the pipeline. Note that thanks to the pipeline structure we can optimize the interaction of the imputation method with the parameters of the downstream classifier without cheating:
In [82]:
%%time
params = {
'imp__strategy': ['mean', 'median'],
'clf__max_features': [0.5, 1],
'clf__max_depth': [3, 4, 5],
}
gs = GridSearchCV(pipeline, params, cv=5, scoring='roc_auc', n_jobs=4)
gs.fit(X_train, y_train)
In [83]:
sorted(gs.grid_scores_, key=lambda x: x.mean_validation_score, reverse=True)
Out[83]:
In [84]:
gs.best_score_
Out[84]:
In [85]:
plot_roc_curve(y_test, gs.predict_proba(X_test))
In [86]:
gs.best_params_
Out[86]:
From this search we can conclude that the imputation by the 'mean' strategy is generally a slightly better imputation strategy when training a GBRT model on this data.
Helper tool for better sklearn / pandas integration: https://github.com/paulgb/sklearn-pandas by making it possible to embed the feature construction from the raw dataframe directly inside a pipeline.
Thanks to:
Kaggle for setting up the Titanic challenge.
This blog post by Philippe Adjiman for inspiration:
http://www.philippeadjiman.com/blog/2013/09/12/a-data-science-exploration-from-the-titanic-in-r/