View source on GitHub |
TF-Ranking BERT task.
Classes
class TFRBertConfig
: The tf-ranking BERT task config.
class TFRBertDataConfig
: Data config for TFR-BERT task.
class TFRBertDataLoader
: A class to load dataset for TFR-BERT task.
class TFRBertModelBuilder
: Model builder for TFR-BERT models.
class TFRBertModelConfig
: A TFR-BERT model configuration.
class TFRBertScorer
: Univariate BERT-based scorer.
class TFRBertTask
: Task object for tf-ranking BERT.
Type Aliases
Other Members | |
---|---|
DOCUMENT_ID |
'document_id'
|
QUERY_ID |
'query_id'
|