torch_frame.utils.infer_df_stype

infer_df_stype(df: DataFrame) dict[str, torch_frame._stype.stype][source]

Infer col_to_stype given DataFrame object.

Parameters:

df (DataFrame) – Input data frame.

Returns:

Inferred col_to_stype, mapping a column name to

its inferred stype.

Return type:

col_to_stype