ML之DT:基于DT算法对泰titanic(泰坦尼克号)数据集进行二分类(是否获救)预测
目录
- X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25, random_state = 33)
-
- vec = DictVectorizer(sparse=False)
- X_train = vec.fit_transform(X_train.to_dict(orient='record'))
- X_test = vec.transform(X_test.to_dict(orient='record'))
-
-
- dtc = DecisionTreeClassifier()
- dtc.fit(X_train, y_train)
- y_predict = dtc.predict(X_test)
- dtc.score(X_test, y_test))
- classification_report(y_predict, y_test, target_names = ['died', 'survived'])
网站声明:如果转载,请联系本站管理员。否则一切后果自行承担。
加入交流群
请使用微信扫一扫!