Type Members

1.  class Encoder extends AnyRef
2.  class EncoderMaxTokensRuntimeException extends RuntimeException
3.  class LinearLayer extends AnyRef

   Implements one linear layer

4.  abstract class LinearLayerFactory extends Sourcer
5.  class LinearLayerFactoryFromFiles extends LinearLayerFactory
6.  class LinearLayerFactoryFromResources extends LinearLayerFactory
7.  class LinearLayerLayout extends AnyRef
8.  class Sourcer extends AnyRef
9.  class TokenClassifier extends AnyRef

   Implements the inference step of a token classifier for multi-task learning The classifier uses a single encoder to generate the hidden state representation for every token and one linear classifier per task to produce task-specific token labels.

   Implements the inference step of a token classifier for multi-task learning The classifier uses a single encoder to generate the hidden state representation for every token and one linear classifier per task to produce task-specific token labels. Note that the encoder is The token classifier expects the model to be loaded to saved in one directory in the following format: encoder.onnx - ONNX-formatted encoder model encoder.name - Hugging Face name of the transformer used as the encoder tasks/ 0/ name - task name weights - weights of the linear classifier biases - biases of the linear classifier labels - array of labels to be predicted 1/ ... (same as task 0) 2/ ... (same as task 0) ...

10.  abstract class TokenClassifierFactory extends Sourcer
11.  class TokenClassifierFactoryFromFiles extends TokenClassifierFactory
12.  class TokenClassifierFactoryFromResources extends TokenClassifierFactory
13.  class TokenClassifierLayout extends AnyRef