def get_column_types()

in src/utils.py [0:0]

• 8 lines of code
• 4 McCabe index (conditional complexity)

def get_column_types(df):
    col_type = {}
    for col in df.columns:
        # If the number of unique items is less than 50 consider them as categorical label
        if df[col].dtype == 'object' or df[col].nunique() < 50:
            col_type[col] = ColType.CATEGORICAL
        else:
            col_type[col] = ColType.NUMERICAL

    return col_type