tmp_file = tmp_file.rename(columns={'类别': 'Category'})
train_resample['Category'] = train_resample.Category.map(lambda x: 1 if x =='水杯' else 0)
