草莓伏特加 发表于 2020-12-11 17:28

如何在python中使用测试集

最近我们老师布置了一个作业,让我们做一个比较基础的深度学习,我的模型已经基本构建完成了。
但是不知道怎么使用测试集与导出预测结果。

我问老师怎么做,他告诉我
# 采用最优参数再次构建逻辑回归模型lr_clf2 = LogisticRegression(penalty="l1", C=1.5, class_weight={0: 0.02, 1: 0.98},solver='liblinear')
lr_clf2.fit(X1_train, y1_train)
y1_train_pred = lr_clf2.predict(X1_train)
print("Confusion matrix (training):\n {0}\n".format(confusion_matrix(y1_train, y1_train_pred)))
print("Classification report (training):\n {0}".format(classification_report(y1_train, y1_train_pred)))

y1_val_pred = lr_clf2.predict(X1_val)
print("Confusion matrix (validation):\n {0}\n".format(confusion_matrix(y1_val, y1_val_pred)))
print("Classification report (validation):\n {0}".format(classification_report(y1_val, y1_val_pred)))

y1_train_pred = lr_clf2.predict(X1_train)这个已经是预测了,但是我还是不明白啊!救救我
页: [1]
查看完整版本: 如何在python中使用测试集