集成算法、特徵選取、泰坦尼克船員獲救案例

筆記

集成算法分類:

  • Bagging:訓練多個分類器取平均,由於各個分類器不相關可同時訓練,例如隨機森林
  • Boosting:根據上一個分類器的結果設置下一次訓練的參數,各個分類器按權重相加,例如Adaboost,Xgboost
  • Stacking:把多個基本算法的預測結果作爲輸入訓練

泰坦尼克船員獲救

數據導入:

import pandas as pd
titanic = pds.read_csv("titanic_train.csv")
# 缺失值處理:用中位數代替
titanic["Age"] = titanic["Age"].fillna(titanic["Age"].median())
# 字符值處理:映射成數字
titanic.loc[titanic["Sex"] == "male", "Sex"] = 0
titanic.loc[titanic["Sex"] == "female", "Sex"] = 1
# 缺失值處理:用出現頻率最高的值代替
titanic["Embarked"] = titanic["Embarked"].fillna('S')
titanic.loc[titanic["Embarked"] == "S", "Embarked"] = 0
titanic.loc[titanic["Embarked"] == "C", "Embarked"] = 1
titanic.loc[titanic["Embarked"] == "Q", "Embarked"] = 2

線性迴歸模型:

from sklearn.linear_model import LinearRegression
from sklearn.model_selection import KFold
import numpy as np

predictors = ["Pclass", "Sex", "Age", "SibSp", "Parch", "Fare", "Embarked"]


alg = LinearRegression()
kf = KFold(n_splits=3)

predictions = []
for train, test in kf.split(titanic):
    train_predictors = (titanic[predictors].iloc[train,:])
    train_target = titanic["Survived"].iloc[train]
    alg.fit(train_predictors, train_target)
    test_predictions = alg.predict(titanic[predictors].iloc[test,:])
    predictions.append(test_predictions)

predictions = np.concatenate(predictions, axis=0)
predictions[predictions > .5] = 1
predictions[predictions <=.5] = 0
accuracy = sum(predictions == titanic["Survived"]) / len(predictions)
print (accuracy)
# 0.7833894500561167

邏輯迴歸模型:

from sklearn.model_selection import cross_val_score
from sklearn.linear_model import LogisticRegression
alg = LogisticRegression(max_iter = 3000)
scores = cross_val_score(alg, titanic[predictors], titanic["Survived"], cv=3)
print(scores.mean())
# 0.7946127946127947

隨機森林模型:

from sklearn.ensemble import RandomForestClassifier
alg = RandomForestClassifier(n_estimators=10, min_samples_split=2, min_samples_leaf=1)
scores = cross_val_score(alg, titanic[predictors], titanic["Survived"], cv=3)
print(scores.mean())
# 0.7934904601571269

# 改變參數值
alg = RandomForestClassifier(random_state=1, n_estimators=100, min_samples_split=4, min_samples_leaf=2)
scores = cross_val_score(alg, titanic[predictors], titanic["Survived"], cv=3)
print(scores.mean())
# 0.8148148148148148

增加一維特徵“頭銜”,從名字中獲取:

import re
def get_title(name):
    title_search = re.search(' ([A-Za-z]+)\.', name)
    if title_search:
        return title_search.group(1)
    return ""

titles = titanic["Name"].apply(get_title)
title_mapping = {"Mr": 1, "Miss": 2, "Mrs": 3, "Master": 4, "Dr": 5, "Rev": 6, "Major": 7, "Col": 7, "Mlle": 8, "Mme": 8, "Don": 9, "Lady": 10, "Countess": 10, "Jonkheer": 10, "Sir": 9, "Capt": 7, "Ms": 2}
for k,v in title_mapping.items():
    titles[titles == k] = v

titanic["Title"] = titles

檢測每個特徵的重要性:

from sklearn.feature_selection import SelectKBest, f_classif
import matplotlib.pyplot as plt

all_predictors = ["Pclass", "Sex", "Age", "SibSp", "Parch", "Fare", "Embarked","Title"]

selector = SelectKBest(f_classif, k=5)
selector.fit(titanic[all_predictors], titanic["Survived"])
scores = -np.log10(selector.pvalues_)

plt.bar(range(len(predictors)), scores)
plt.xticks(range(len(predictors)), predictors, rotation='vertical')
plt.show()

在這裏插入圖片描述
選出最重要的四個特徵進行stacking算法:

from sklearn.ensemble import GradientBoostingClassifier
predictors = ["Pclass", "Sex", "Fare", "Title"]

algorithms = [
    [GradientBoostingClassifier(n_estimators=25, max_depth=3), predictors],
    [LogisticRegression(max_iter = 3000), predictors]
]

kf = KFold(n_splits=3)

predictions = []
for train, test in kf.split(titanic):
    train_target = titanic["Survived"].iloc[train]
    full_test_predictions = []
    for alg, predictors in algorithms:
        alg.fit(titanic[predictors].iloc[train,:], train_target)
        test_predictions = alg.predict_proba(titanic[predictors].iloc[test,:].astype(float))[:,1]
        full_test_predictions.append(test_predictions)
    test_predictions = (full_test_predictions[0] + full_test_predictions[1]) / 2
    test_predictions[test_predictions <= .5] = 0
    test_predictions[test_predictions > .5] = 1
    predictions.append(test_predictions)


predictions = np.concatenate(predictions, axis=0)
accuracy = sum(predictions == titanic["Survived"]) / len(predictions)
print(accuracy)
# 0.8181818181818182
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章