決定木を利用して分析する。
データの作成
y_train = d_train["Survived"].values x_train = d_train[["Pclass", "Sex", "Age", "Fare", "Parch", "Embarked", "SibSp"]].values x_test = d_test[["Pclass", "Sex", "Age", "Fare", "Parch", "Embarked", "SibSp"]].values
予測モデル作成
from sklearn.tree import DecisionTreeClassifier dtree = DecisionTreeClassifier() dtree.fit(x_train,y_train) predictions = dtree.predict(x_test)
提出ファイル作成
kaggle_submission = pd.DataFrame({
"PassengerId": d_test["PassengerId"],
"Survived": predictions
})
kaggle_submission.to_csv("kaggle_submission.csv", index=False)
