fix 0 score in evaluate (#226)

stephantul · web-flow · commit 2d0417d67571 · 2025-04-28T10:06:54.000+02:00
diff --git a/model2vec/inference/model.py b/model2vec/inference/model.py
@@ -299,8 +299,6 @@ def evaluate_single_or_multi_label(
     report = classification_report(
         y,
         predictions,
-        labels=np.arange(len(classes)),
-        target_names=[str(c) for c in classes],
         output_dict=output_dict,
         zero_division=0,
     )

Original file line number	Diff line number	Diff line change
`@@ -299,8 +299,6 @@ def evaluate_single_or_multi_label(`
`299`	`299`	`report = classification_report(`
`300`	`300`	`y,`
`301`	`301`	`predictions,`
`302`		`- labels=np.arange(len(classes)),`
`303`		`- target_names=[str(c) for c in classes],`
`304`	`302`	`output_dict=output_dict,`
`305`	`303`	`zero_division=0,`
`306`	`304`	`)`