閑古鳥

オールドプログラマの日記。プログラミングとか病気(透析)の話とか。

混合行列わからん!

Confusion Matrixって混乱行列じゃん! 左右盲なのかわからないけど、右左も間違えるのにこんなの覚えられませんわ。

というわけでsklearnに全部やってもらう。

from sklearn.metrics import confusion_matrix, accuracy_score, precision_score, recall_score, f1_score

# 混合行列を求める
y_true = [1, 1, 1, 0, 0, 0]
y_pred = [1, 0, 0, 1, 0, 0]
print(confusion_matrix(y_true, y_pred))

# そもそもprecisionはどうするんだっけ?とか考えなくてもいい
print(accuracy_score(y_true, y_pred))
print(precision_score(y_true, y_pred))
print(recall_score(y_true, y_pred))
print(f1_score(y_true, y_pred))

実装はこれでよくても、実際にやっていくにはやっぱりちゃんと理解しないといけないのでしょうけど、なかなかカンペ見ながらでないと混乱してしまいますね。