Я пытаюсь получить 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