Dataset#

class gpjax.dataset.Dataset(X=None, y=None)[source]#

Bases: object

Base class for datasets.

Parameters:
  • X (Num[jaxlib._jax.Array, 'N D'] | Num[ndarray, 'N D'] | None) – input data.

  • y (Num[jaxlib._jax.Array, 'N Q'] | Num[ndarray, 'N Q'] | None) – output data.

Expand for references to gpjax.dataset.Dataset

Likelihood guide / Likelihoods in GPJax

fit

get_batch

ConjugatePosterior

create_oilmm_from_data

conjugate_loocv

StateSpaceConjugatePosterior

fit

fit_lbfgs

fit_scipy

state_space_mll

property in_dim: int#

Dimension of the inputs, \(X\).

is_supervised()[source]#

Returns True if the dataset is supervised.

Return type:

bool

is_unsupervised()[source]#

Returns True if the dataset is unsupervised.

Return type:

bool

property multi_output: bool#

True if the dataset has more than one output dimension.

property n: int#

Number of observations.

Expand for references to gpjax.dataset.Dataset.n

fit

ConjugatePosterior

conjugate_loocv

StateSpaceConjugatePosterior

property num_outputs: int#

Number of output dimensions.