本文對Kaggle中的Titanic事故中乘客遇難情況進(jìn)行了相應(yīng)的分析和可視化采用邏輯回歸對他們的遇難情況進(jìn)行了預(yù)測。最后得到的預(yù)測結(jié)果不算很好但是本文大致是一個較為完整的數(shù)據(jù)分析和預(yù)測流程。
建立模型時我們需要的特征往往都是數(shù)值型因此我們需要將對相應(yīng)屬性轉(zhuǎn)換為one-hot編碼表示首先我們打印出數(shù)據(jù)集的前5行看看哪些特征需要進(jìn)行轉(zhuǎn)換
將模型中每個特征的系數(shù)打印出來分析可以看到頭等艙可以很好的提升獲救率而三等艙與獲救率則有著明顯的負(fù)相關(guān)女性與獲救率之間則有著明顯的正相關(guān)等。
之前我們直接在測試集上進(jìn)行了預(yù)測為了優(yōu)化模型我們可以首先在一部分訓(xùn)練集上進(jìn)行預(yù)測與真實(shí)的情況進(jìn)行對比采用sklearn里的交叉預(yù)測模塊進(jìn)行分析
在這之后為了提高預(yù)測準(zhǔn)確率還需要進(jìn)行更進(jìn)一步的特征工程的工作并且采用多模型融合也可能可以提高預(yù)測的準(zhǔn)確率。本文在補(bǔ)全未知數(shù)據(jù)時采用了較為簡單的方法尤其時對年齡的補(bǔ)全采用了所有數(shù)據(jù)的平均值作為缺失年齡的補(bǔ)全值這樣的做法可能并不科學(xué)另外對于PassengerId、Name和Ticket的信息沒有進(jìn)行有效的利用在這些方面進(jìn)一步深挖也許可以提高預(yù)測的準(zhǔn)確率。