MCPcopy Create free account
hub / github.com/allenai/scibert / BertTextClassifier

Class BertTextClassifier

scibert/models/bert_text_classifier.py:16–90  ·  view source on GitHub ↗

Implements a basic text classifier: 1) Embed tokens using `text_field_embedder` 2) Get the CLS token 3) Final feedforward layer Optimized with CrossEntropyLoss. Evaluated with CategoricalAccuracy & F1.

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected