Package chat.octet.model.components.criteria

Class StoppingCriteriaList

java.lang.Object

java.util.AbstractCollection<E>

java.util.AbstractList<E>

java.util.ArrayList<StoppingCriteria>

chat.octet.model.components.criteria.StoppingCriteriaList

All Implemented Interfaces:

StoppingCriteria, Serializable, Cloneable, Iterable<StoppingCriteria>, Collection<StoppingCriteria>, List<StoppingCriteria>, RandomAccess

public class StoppingCriteriaList
extends ArrayList<StoppingCriteria>
implements StoppingCriteria

Stopping criteria list

Author:

William

See Also:

• Serialized Form

Field Summary

Fields inherited from class java.util.AbstractList

modCount

Constructor Summary

Constructors

Constructor	Description
StoppingCriteriaList()
StoppingCriteriaList(Collection<? extends StoppingCriteria> c)

Method Summary

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

Methods inherited from class java.util.ArrayList

add, add, addAll, addAll, clear, clone, contains, ensureCapacity, equals, forEach, get, hashCode, indexOf, isEmpty, iterator, lastIndexOf, listIterator, listIterator, remove, remove, removeAll, removeIf, removeRange, replaceAll, retainAll, set, size, sort, spliterator, subList, toArray, toArray, trimToSize

Methods inherited from class java.util.AbstractCollection

containsAll, toString

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface java.util.Collection

parallelStream, stream, toArray

Methods inherited from interface java.util.List

containsAll

Constructor Details

StoppingCriteriaList

public StoppingCriteriaList()

StoppingCriteriaList

public StoppingCriteriaList(Collection<? extends StoppingCriteria> c)

Method Details

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.