任务3 - 建模
用逻辑回归、svm和决策树;随机森林和XGBoost进行模型构建,评分方式任意,如准确率等。(不需要考虑模型调参)
一、 使用逻辑回归
from sklearn.linear_model import SGDClassifier
1 初始化模型
log_reg = SGDClassifier(loss='log',penalty='l2',max_iter = 200)
2 训练
log_reg.fit(X_train, y_train)
3 预测
y_pred_lgr = log_reg.predict_proba(X_test)
print (y_pred_lgr[:5])
4 评估
from sklearn.metrics import accuracy_score
from sklearn.metrics import precision_recall_fscore_support
from sklearn.metrics import classification_report,confusion_matrix