EL之DT&RF&GBT:基于三种算法(DT、RF、GBT)对titanic(泰坦尼克号)乘客数据集进行二分类(是否获救)预测并对比各自性能
目录
- 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.fit(X_train, y_train)
- dtc_y_pred = dtc.predict(X_test)
-
-
- rfc.fit(X_train, y_train)
- rfc_y_pred = rfc.predict(X_test)
-
- rfc.score(X_test, y_test)
- classification_report(rfc_y_pred, y_test)
-
-
- gbc.fit(X_train, y_train)
- gbc_y_pred = gbc.predict(X_test)
-
- gbc.score(X_test, y_test)
- classification_report(gbc_y_pred, y_test)
网站声明:如果转载,请联系本站管理员。否则一切后果自行承担。
加入交流群
请使用微信扫一扫!