Skip to content

Commit de6cbf2

Browse files
committed
update cost function
1 parent f314970 commit de6cbf2

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

redis_retrieval_optimizer/bayes_study.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -152,10 +152,12 @@ def objective(trial, study_config, redis_url, corpus_processor, search_method_ma
152152
search_method_output = search_fn(search_input)
153153

154154
trial_metrics = utils.eval_trial_metrics(qrels, search_method_output.run)
155-
trial_metrics["total_indexing_time"] = total_indexing_time
156-
trial_metrics["avg_query_time"] = utils.get_query_time_stats(
157-
search_method_output.query_metrics.query_times
158-
)["avg_query_time"]
155+
trial_metrics["total_indexing_time"] = -(total_indexing_time)
156+
trial_metrics["avg_query_time"] = -(
157+
utils.get_query_time_stats(search_method_output.query_metrics.query_times)[
158+
"avg_query_time"
159+
]
160+
)
159161

160162
# save results as we go in case of failure
161163
persist_metrics(redis_url, trial_settings, trial_metrics, study_config.study_id)

0 commit comments

Comments
 (0)