Log number of spam/non-spam bugs when training the SpamBug model (#1214)

Fixes #1201

Co-authored-by: Marco Castelluccio <mcastelluccio@mozilla.com>
This commit is contained in:
Pranay Khaire 2020-01-03 22:15:08 +05:30 коммит произвёл Marco Castelluccio
Родитель 1752c8ed25
Коммит 40e6221b4a
1 изменённых файлов: 11 добавлений и 0 удалений

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

@ -85,6 +85,17 @@ class SpamBugModel(BugModel):
):
classes[bug_id] = 1
print(
"{} bugs are classified as non-spam".format(
sum(1 for label in classes.values() if label == 0)
)
)
print(
"{} bugs are classified as spam".format(
sum(1 for label in classes.values() if label == 1)
)
)
return classes, [0, 1]
def items_gen(self, classes):