您的位置:首页 > 其它

kaggle 泰坦尼克号生还者预测

2017-09-28 17:07 363 查看
import pandas as pd
from sklearn.tree import DecisionTreeClassifier #决策树
from sklearn.model_selection import cross_val_score
df = pd.read_csv("train.csv")
#数据清洗,补全有缺失的数据
df.Age.fillna(df.Age.mean(),inplace = True)
#将female,male替换成数字1 0
df["Sex_int"] = df["Sex"].map({"female":1 , "male":0})
# print(df.info())
#选择特征
X = df[["Sex_int", "Age" , "Pclass"]]
y = df.Survived
model = DecisionTreeClassifier()
scores = cross_val_score(model , X , y , cv= 10 )
print(scores)
print(scores.mean())
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: