RegressionDataset

A borrowed tensor dataset for regression tasks.

Instance members

NameDescription
this.FeatureCountNumber of features per example.
this.FeaturesFeature matrix with shape [n, f] and dtype float32.
this.LabelsTarget values with shape [n] and dtype float32.
this.RowCountNumber of examples.

RegressionDataset module

Constructors for regression datasets.

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

create

create features labels

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

Parameters

  • features : Tensor
  • labels : Tensor

Returns RegressionDataset