Skip to content

Commit c4e2de9

Browse files
authored
Fix a corner case in h-cls greedy resolver (#204)
* Fix a corner case in h-cls greedy resolver * Prevent greedy resolver to insert parents which were not predicted * Revert "Prevent greedy resolver to insert parents which were not predicted" This reverts commit 0eb8ec6.
1 parent 5a06e0c commit c4e2de9

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

model_api/cpp/models/src/classification_model.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -606,7 +606,7 @@ std::map<std::string, float> GreedyLabelsResolver::resolve_labels(const std::vec
606606

607607
std::vector<std::string> candidates;
608608
for (const auto& g : label_groups) {
609-
if (g.size() == 1) {
609+
if (g.size() == 1 && label_to_prob[g[0]] > 0.f) {
610610
candidates.push_back(g[0]);
611611
}
612612
else {

model_api/python/model_api/models/classification.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,7 +410,7 @@ def get_predecessors(lbl, candidates):
410410

411411
candidates = []
412412
for g in self.label_groups:
413-
if len(g) == 1:
413+
if len(g) == 1 and label_to_prob[g[0]] > 0.0:
414414
candidates.append(g[0])
415415
else:
416416
max_prob = 0.0

0 commit comments

Comments
 (0)