botiverse.models.LinearClassifier package

Submodules

botiverse.models.LinearClassifier.LinearClassifier module

class botiverse.models.LinearClassifier.LinearClassifier.LinearClassifier(input_embeddings_size, output_size)

Bases: Module

An interface for a simple linear classifier intended to be used with the BERT model

Initializes internal Module state, shared by both nn.Module and ScriptModule.

forward(input)

Defines the computation performed by the model.

Parameters:

input (Tensor) – The provided input from the BERT model.

Returns:

Output after the forward pass (classes probabilities).

Return type:

Tensor

training: bool

Module contents