Package chat.octet.model.components.criteria

Class StoppingCriteriaList

java.lang.Object

chat.octet.model.components.criteria.StoppingCriteriaList

All Implemented Interfaces:

StoppingCriteria

public class StoppingCriteriaList
extends Object
implements StoppingCriteria

Stopping criteria list

Author:

William

Constructor Summary

Constructors

Constructor	Description
StoppingCriteriaList()

Method Summary

Modifier and Type	Method	Description
StoppingCriteriaList	add(StoppingCriteria criteria)
boolean	criteria(int[] inputTokenIds, float[] scores, Object... args)	Stopping criteria
boolean	isEmpty()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

StoppingCriteriaList

public StoppingCriteriaList()

Method Details

add

public StoppingCriteriaList add(StoppingCriteria criteria)

isEmpty

public boolean isEmpty()

criteria

public boolean criteria(@Nullable
 int[] inputTokenIds,
 @Nonnull
 float[] scores,
 Object... args)

Description copied from interface: StoppingCriteria

Stopping criteria

Specified by:

criteria in interface StoppingCriteria

Parameters:

inputTokenIds - Indices of input sequence tokens in the vocabulary.

scores - Prediction scores of a language modeling head. These can be logits for each vocabulary.

args - Specific args to a stopping criteria.

Returns:

boolean `False` indicates we should continue, `True` indicates we should stop.