df2ds

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

Converts DataFrame to Dataset

Parameters:

Name	Type	Description	Default
df	DataFrame	input DataFrame	required

Returns:

Type	Description
Dataset	dataset

Source code in mono_dense_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