Class: Informers::DistilBertForSequenceClassification

Inherits:

DistilBertPreTrainedModel

• Object
• PreTrainedModel
• DistilBertPreTrainedModel
• Informers::DistilBertForSequenceClassification

Defined in:

lib/informers/models.rb

Instance Attribute Summary

Attributes inherited from PreTrainedModel

#config

Instance Method Summary

• #call(model_inputs) ⇒ Object

Methods inherited from PreTrainedModel

construct_session, from_pretrained, #initialize

Constructor Details

This class inherits a constructor from Informers::PreTrainedModel

Instance Method Details

#call(model_inputs) ⇒ Object

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

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