RankingDataset

A borrowed tensor dataset for learning-to-rank tasks.

Instance members

NameDescription
this.FeatureCountNumber of features per example.
this.FeaturesFeature matrix with shape [n, f] and dtype float32.
this.GroupIdsQuery identifiers with shape [n] and dtype int64.
this.LabelsRelevance labels with shape [n] and dtype float32.
this.RowCountNumber of examples.

RankingDataset module

Constructors for learning-to-rank datasets.

FunctionDescription
create features labels groupIdsCreate a dataset without copying or taking ownership of its tensors.

create

create features labels groupIds

Create a dataset without copying or taking ownership of its tensors.

Parameters

  • features : Tensor
  • labels : Tensor
  • groupIds : Tensor

Returns RankingDataset