本文接前文Kaggle入門，主要的區別是更換了分類器XGBoost。

# -*- coding: utf-8 -*-import pandas as pd import xgboost as xgb# 讀訓練數據 data = pd.read_csv('data/train.csv')#數據預處理 data['Sex'] = data['Sex'].apply(lambda s: 1 if s == 'male' else 0) #把性別從字符串類型轉換為0或1數值型數據 data = data.fillna(0) #缺失字段填0 # 選取特征 X_train = data[['Sex', 'Age', 'Pclass', 'SibSp', 'Parch', 'Fare']].as_matrix() #字段說明：性別，年齡，客艙等級，兄弟姐妹和配偶在船數量，父母孩子在船的數量，船票價格# 建立標簽數據集 y_train = data['Survived']#訓練模型 model = xgb.XGBClassifier(max_depth=10, n_estimators=300, learning_rate=0.01).fit(X_train, y_train)# 讀測試數據 testdata = pd.read_csv('data/test.csv')#數據清洗, 數據預處理 testdata = testdata.fillna(0) testdata['Sex'] = testdata['Sex'].apply(lambda s: 1 if s == 'male' else 0)#特征選擇 X_test = testdata[['Sex', 'Age', 'Pclass', 'SibSp', 'Parch', 'Fare']].as_matrix()#評估模型 predictions = model.predict(X_test)# 保存預測結果 submission = pd.DataFrame({ 'PassengerId': testdata['PassengerId'],'Survived': predictions }) submission.to_csv("titanic_xgboost_submission.csv", index=False)

總結

以上是生活随笔為你收集整理的Kaggle入门 (Titanic XGBoost)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。