Package chat.octet.model.components.criteria.impl

Class StoppingWordCriteria

java.lang.Object

chat.octet.model.components.criteria.impl.StoppingWordCriteria

All Implemented Interfaces:

StoppingCriteria

public class StoppingWordCriteria
extends Object
implements StoppingCriteria

Constructor Summary

Constructors

Constructor	Description
StoppingWordCriteria(String... words)

Method Summary

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

Methods inherited from class java.lang.Object

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

Constructor Details

StoppingWordCriteria

public StoppingWordCriteria(String... words)

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.