search

taokz / BiomedGPT

BiomedGPT: A Unified and Generalist Biomedical Generative Pre-trained Transformer for Vision, Language, and Multimodal Tasks
Apache License 2.0
363 stars 34 forks source link

"_class2label.pkl" file for classification task is missing #5

Closed wafaAlghallabi closed 10 months ago

wafaAlghallabi commented 10 months ago

Hello,

Thank you very much for providing the code.

I'm trying to use the finetune script on MedMNIST datasets to reproduce the results reported in the paper but "bloodmnist_class2label.pkl" is missing. May I ask you how should I generate this file, please?

Thanks for your time.

taokz commented 10 months ago

There is the class2labl information in scripts/preprocess/finetuning/medmnist.py, you can use the INFO_DICT['bloodmnist']['label'], and save it to be pkl.