Skip to contents

Function for converting an arrow data set into a data set that can be used to store and process embeddings.

Usage

py_dataset_to_embeddings(py_dataset)

Arguments

py_dataset

Object of class datasets.arrow_dataset.Dataset.

Value

Returns the data set of class datasets.arrow_dataset.Dataset with only two columns ("id","input"). "id" stores the name of the cases while "input" stores the embeddings.