df2ds

airt.keras.experiments.df2ds(df: pd.DataFrame) -> tf.data.Dataset

Converts DataFrame to Dataset

Parameters:

Name	Type	Description	Default
df	pd.DataFrame	input DataFrame	required

Returns:

Type	Description
tf.data.Dataset	dataset

Source code in airt/keras/experiments.py

def df2ds(df: pd.DataFrame) -> tf.data.Dataset:
    """Converts DataFrame to Dataset

    Args:
        df: input DataFrame

    Returns:
        dataset
    """
    x = df.to_dict("list")
    y = x.pop("ground_truth")

    ds = tf.data.Dataset.from_tensor_slices((x, y))

    return ds