mbrs.decoders.rerank module

mbrs.decoders.rerank module#

class mbrs.decoders.rerank.DecoderRerank(cfg: ~mbrs.decoders.base.DecoderBase.Config, metric: ~mbrs.metrics.base.MetricBase, selector: ~mbrs.selectors.base.Selector = <mbrs.selectors.nbest.SelectorNbest object>)[source]#

Bases: DecoderReferenceless

Reranking decoder class.

  • Time complexity: O(N)

  • Space complexity: O(N)

decode(hypotheses: list[str], source: str, nbest: int = 1) → Output[source]#

Select the n-best hypotheses based on the strategy.

Parameters:

  • hypotheses (list[str]) – Hypotheses.

  • source (str) – A source.

  • nbest (int) – Return the n-best hypotheses.

Returns:

The n-best hypotheses.

Return type:

DecoderRerank.Output

Contents