推測になるのですが、df_train[i:i+6]
したときに、切り取られる区間によってはY_train
, Y_pred
の両方ともが全て 1 になってしまうケースが発生しているのではないでしょうか。
その場合、cm = confusion_matrix(Y_train,Y_pred)
は1x1の配列が返ってくるので(1ですべて正解)、次のtp,fn,fp,tn=cm.flatten()
で 4つの値に unpack できず、質問にあるエラーがでています。
confusion_matrix()
で、明示的にlabels
を指定してやることで、この問題は回避できると思います。(tp, fn, fp, tn の順番がこれであっているかは、ご自身で確認お願いします)
python
1 cm = confusion_matrix(Y_train, Y_pred, labels=[1, -1])
2 tp, fn, fp, tn = cm.flatten()
あと、本題とは関係ない部分ですが、7行ずつ取り出したいなら、df_train[i:i+6]
ではなくdf_train[i:i+7]
になると思います。
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2022/07/19 14:39