How do I generate the ‘text_features.pkl’ file

HestiaSky / IMF-Pytorch

MIT License

34 stars 5 forks source link

How do I generate the ‘text_features.pkl’ file #3

Closed aduaa closed 1 year ago

aduaa commented 1 year ago

Could you please tell me how the text_features.pkl file is generated

xiaohetongxue commented 1 year ago

Have you solved this problem? I also ran into this problem, wondering how the text_featues.pkl file is generated?

HestiaSky commented 1 year ago

Hi, thanks for your interest. text_features.pkl is generated by the pool output of BERT-base-uncased model of the text descriptions. You can use the open-released model from huggingface and the text descriptions are contained in the 'dataset' folder.