reagent.prediction.ranking package

Submodules

reagent.prediction.ranking.predictor_wrapper module

class reagent.prediction.ranking.predictor_wrapper.DeterminantalPointProcessPredictorWrapper(alpha: float, kernel: reagent.prediction.ranking.predictor_wrapper.Kernel = Kernel.Linear, sigma: float = 1.0, rerank_topk: Optional[int] = None)

Bases: torch.jit._script.ScriptModule

http://jgillenw.com/cikm2018.pdf Algorithm 1

forward: Callable[[...], Any]
greedy_select(L)
training: bool
unchosen_dets(L, chosen: List[int])
class reagent.prediction.ranking.predictor_wrapper.Kernel(value)

Bases: enum.Enum

An enumeration.

Linear = 'linear'
RBF = 'rbf'

Module contents