Class: Informers::BertForSequenceClassification
- Inherits:
-
BertPreTrainedModel
- Object
- PreTrainedModel
- BertPreTrainedModel
- Informers::BertForSequenceClassification
- Defined in:
- lib/informers/models.rb
Constant Summary
Constants inherited from PreTrainedModel
PreTrainedModel::MAIN_INPUT_NAME
Instance Attribute Summary
Attributes inherited from PreTrainedModel
Instance Method Summary collapse
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
775 776 777 |
# File 'lib/informers/models.rb', line 775 def call(model_inputs) SequenceClassifierOutput.new(*super(model_inputs)) end |