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'
|