Class: Informers::BertForTokenClassification

Inherits:

BertPreTrainedModel

• Object
• PreTrainedModel
• BertPreTrainedModel
• Informers::BertForTokenClassification

Defined in:

lib/informers/models.rb

Constant Summary

Constants inherited from PreTrainedModel

PreTrainedModel::MAIN_INPUT_NAME

Instance Attribute Summary

Attributes inherited from PreTrainedModel

#config

Instance Method Summary

• #call(model_inputs) ⇒ Object

Methods inherited from PreTrainedModel

construct_session, from_pretrained, #generate, #initialize

Constructor Details

This class inherits a constructor from Informers::PreTrainedModel

Instance Method Details

#call(model_inputs) ⇒ Object

# File 'lib/informers/models.rb', line 781

def call(model_inputs)
  TokenClassifierOutput.new(*super(model_inputs))
end