Evaluate the model with 0.85, 0.9, 0.95 confidence thresholds too

Part of #1525
This commit is contained in:
Marco Castelluccio 2020-05-03 13:25:35 +02:00
Родитель e8e760d892
Коммит 4608f3c31b
1 изменённых файлов: 1 добавлений и 1 удалений

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

@ -405,7 +405,7 @@ class TestSelectModel(Model):
for cap in [None, 300, 500]:
for reduction in reductions:
for confidence_threshold in [0.5, 0.7, 0.8]:
for confidence_threshold in [0.5, 0.7, 0.8, 0.85, 0.9, 0.95]:
do_eval(confidence_threshold, reduction, cap)
def get_feature_names(self):