Получение «индекса 1 выходит за пределы» при попытке получить False Negative в Python - PullRequest
0 голосов
/ 20 апреля 2020

Я пытаюсь получить TP, TN, FP и FN для моего прогноза

Это мой код

xtrain, xval, ytrain, yval = train_test_split(df_features, df_results, test_size=0.2, random_state=9)


clf = RandomForestClassifier(n_estimators=1000, max_depth=10, random_state=0)

clf.fit(xtrain, ytrain)

y_pred = clf.predict(xval)

CM = confusion_matrix(yval, y_pred)

TN = CM[0][0]
FN = CM[1][0]
TP = CM[1][1]
FP = CM[0][1]

Я получаю значение TN, но когда я пытаюсь получить значение FN Я получаю эту ошибку

CM[1][0]
Traceback (most recent call last):

  File "<ipython-input-18-4e0323bdbf74>", line 1, in <module>
    CM[1][0]

IndexError: index 1 is out of bounds for axis 0 with size 1

Как я могу это исправить?

Редактировать:

print(CM)
[[19]]

вот так выглядит yval как

yval

6     A72
46    B83
26    B88
76    A94
28    B99
38    A72
61    A72
3     B88
20    A29
45    C52
86    A75
...