python - 如何使用 f1_score 为每个班级添加标签？

Question

pred = model.predict(X_test)
indexes = tf.argmax(pred, axis=1)


f1_score(y_test, indexes, average=None)

result:

array([0.        , 0.        , 0.        , 0.        , 0.        ,
       0.        , 0.        , 0.        , 0.        , 0.        ,
       0.        , 0.        , 0.        , 0.        , 0.        ,
       0.        , 0.        , 0.        , 0.        , 0.        ,
       0.        , 0.        , 0.08695652, 0.        , 0.        ,
       0.        , 0.        , 0.        , 0.09090909, 0.        ,
       0.        , 0.        , 0.        , 0.        , 0.        ,
       0.        , 0.        , 0.        , 0.        , 0.        ,
       0.        , 0.        , 0.        , 0.        , 0.        ,
       0.        , 0.        , 0.        , 0.        , 0.        ,
       0.        , 0.        , 0.        , 0.        , 0.        ,

我有 256 个班级，我想用索引显示每个班级。1:0, 2:0, 3:0, 4:0.6 像这样

标签： pythonsklearn-pandas