org.clulab.scala_transformers.encoder
Type members
Classlikes
Implements one linear layer
Implements one linear layer
- Companion:
- object
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) ...
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) ...
- Companion:
- object
- Companion:
- object