划分训练集和测试集 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=0)怎样删除测试集中多个特定预测值对应行
时间: 2023-07-14 15:14:29 浏览: 73
一个线性回归模型实例,我们使用train-test-split函数将数据集拆分为训练集和测试集
可以使用 Pandas 库中的 drop() 函数删除测试集中特定预测值对应的行。假设测试集的预测结果保存在 y_pred 变量中,特定预测值为 1,代码如下:
```
import pandas as pd
df_test = pd.DataFrame({'X': X_test, 'y': y_test, 'y_pred': y_pred})
df_test = df_test[df_test['y_pred'] != 1]
X_test = df_test['X']
y_test = df_test['y']
```
首先将测试集的数据、真实标签和预测结果保存到一个 DataFrame 中,然后根据 y_pred 列的值筛选出不等于 1 的行,最后重新赋值给 X_test 和 y_test 变量即可。
阅读全文