kedro_datasets_experimental

Description

Classes

darts.DartsTorchModelDataset(*, filepath, ...)

DartsTorchModelDataset loads and saves Darts TorchForecastingModel instances.

databricks.ExternalTableDataset(*, table[, ...])

ExternalTableDataset loads and saves data into external tables in Databricks.

langchain.ChatAnthropicDataset(credentials)

ChatAnthropicDataset loads a ChatAnthropic langchain model.

langchain.ChatCohereDataset(credentials[, ...])

langchain.ChatOpenAIDataset(credentials[, ...])

langchain.OpenAIEmbeddingsDataset(credentials)

netcdf.NetCDFDataset(*, filepath[, ...])

NetCDFDataset loads/saves data from/to a NetCDF file using an underlying filesystem (e.g.: local, S3, GCS).

optuna.StudyDataset(*, backend, database, ...)

StudyDataset loads/saves data from/to an optuna Study.

prophet.ProphetModelDataset(*, filepath[, ...])

ProphetModelDataset loads/saves Facebook Prophet models to a JSON file using an underlying filesystem (e.g., local, S3, GCS).

pytorch.PyTorchDataset(*, filepath[, ...])

PyTorchDataset loads and saves PyTorch models' state_dict using PyTorch's recommended zipfile serialization protocol.

rioxarray.GeoTIFFDataset(*, filepath[, ...])

GeoTIFFDataset loads and saves rasterdata files and reads them as xarray DataArrays.

safetensors.SafetensorsDataset(*, filepath)

SafetensorsDataset loads/saves data from/to a Safetensors file using an underlying filesystem (e.g.: local, S3, GCS).

video.VideoDataset

Special type indicating an unconstrained type.