Source code for btb.tuning.acquisition.predicted_score

# -*- coding: utf-8 -*-

from btb.tuning.acquisition.base import BaseAcquisition


[docs]class PredictedScoreAcquisition(BaseAcquisition):

    def _acquire(self, candidates, num_candidates=1):
        candidates = candidates if candidates.ndim == 1 else candidates[:, 0]
        return self._get_max_candidates(candidates, num_candidates)