Fix calculation of actually classified elements (#1006)

This commit is contained in:
Marco 2019-10-04 15:58:53 +01:00 коммит произвёл GitHub
Родитель 738435e7cb
Коммит bb652ef26c
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 3 добавлений и 1 удалений

Просмотреть файл

@ -486,8 +486,10 @@ class Model:
np.array(y_pred_filter[classified_indices], dtype=int)
)
classified_num = sum(1 for v in y_pred_filter if v != "__NOT_CLASSIFIED__")
print(
f"\nConfidence threshold > {confidence_threshold} - {len(y_test)} classified"
f"\nConfidence threshold > {confidence_threshold} - {classified_num} classified"
)
if is_multilabel:
confusion_matrix = metrics.multilabel_confusion_matrix(