fpr, tpr, thresholds = roc_curve(y_test_binary[:,0], y_pred)报错 fpr, tpr, thresholds = roc_curve(y_test_binary[:,0], y_pred) IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed
时间: 2023-06-16 12:03:22 浏览: 179
ROC-curve.rar_open_roc_roc_curve
这个错误通常是由于 `y_test_binary` 参数的维度不正确导致的。根据错误提示,`y_test_binary` 是一个一维数组,但是你尝试使用 `y_test_binary[:,0]` 对其进行切片,这是不合法的。
`roc_curve` 函数需要接受两个一维数组,分别表示真实标签和预测标签的概率。你可以尝试检查一下 `y_test_binary` 的形状,并确保它是一个一维数组。如果不是,你可以使用 `numpy` 库中的 `ravel` 函数将其转换为一维数组。例如:
```python
import numpy as np
from sklearn.metrics import roc_curve
y_test_binary = np.ravel(y_test_binary) # 将 y_test_binary 转换为一维数组
fpr, tpr, thresholds = roc_curve(y_test_binary, y_pred)
```
另外,你还需要确保 `y_pred` 参数的形状与 `y_test_binary` 相同,即它们包含相同数量的元素。
阅读全文