I want to use this code to run on USPS dataset,how can i change the dataset , i always meet a typeerror. wait for your answer and thank you!