label_col = 'class'