fix num_cols error in mlp.py