二分类classification是1和2 怎么改成0和1
时间: 2024-02-25 08:56:42 浏览: 28
可以使用sklearn中的LabelEncoder将1和2转换成0和1,示例如下:
```python
from sklearn.preprocessing import LabelEncoder
# 生成示例数据
y_true = [1, 2, 1, 2]
# 将1和2转换成0和1
le = LabelEncoder()
y_true = le.fit_transform(y_true)
print(y_true)
```
输出:
```
[0 1 0 1]
```
这样就可以将1和2转换成0和1,方便后续使用二分类模型进行训练和评估。