Skip to content

Commit a61c7d4

Browse files
ClarkChin08ftian1
authored andcommitted
[fix] fix objective compare logic
1 parent 00d6e55 commit a61c7d4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

lpot/objective.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -145,7 +145,8 @@ def __init__(self, accuracy_criterion, is_measure=False):
145145
self.higher_is_better = True
146146
for k, v in accuracy_criterion.items():
147147
if k in ['relative', 'absolute']:
148-
assert float(v) < 1 and float(v) > -1
148+
if k == 'relative':
149+
assert float(v) < 1 and float(v) > -1
149150
self.relative = True if k == 'relative' else False
150151
self.acc_goal = float(v)
151152
elif k == 'higher_is_better':
@@ -181,7 +182,7 @@ def compare(self, last, baseline):
181182
else base_acc + float(self.acc_goal)
182183

183184
if last_measure == 0 or perf < last_measure:
184-
return acc < acc_target if self.higher_is_better else acc >= acc_target
185+
return acc >= acc_target if self.higher_is_better else acc < acc_target
185186
else:
186187
return False
187188

0 commit comments

Comments
 (0)