How to use the method on a custom dataset?

pietrobarbiero / pytorch_explain

PyTorch Explain: Interpretable Deep Learning in Python.

Apache License 2.0

135 stars 12 forks source link

How to use the method on a custom dataset? #7

Closed bakachan19 closed 12 months ago

bakachan19 commented 1 year ago

Hi. thanks for such an interesting work. I was wondering if your method can be applied to a custom image dataset. I've seen that in the paper "Concept Embedding Models: Beyond the Accuracy-Explainability Trade-Off" you also did experiments on CUB dataset but I could not find the associated code.

Can you please provide the code for the CUB dataset or give some instructions on how to prepare a custom dataset of images in order to apply your method?

Thank you. Best wishes.

mateoespinosa commented 12 months ago

Hey @bakachan19! Thanks for opening this ticket! For some useful examples, please refer to this example on using a CEM or this notebook on the official CEM repository. If you have any issues with those examples and/or code, please reopen this ticket and let us know how we can help. Happy coding and researching!

bakachan19 commented 11 months ago

Dear @mateoespinosa, Thank you for your help. I will give it a try and let you know if I need additional information/help. Have a great day!