如果想要保留三个类别标签而不是本repo.实验中的只有一个类别，需要怎么修改organize_txt_labels?

如果想要保留三个类别标签而不是本repo.实验中的只有一个类别，需要怎么修改organize_txt_labels? #16