search

ejkim47 / prob-cbm

Official code for "Probabilistic Concept Bottleneck Models (ICML 2023)"
MIT License
16 stars 4 forks source link

Scripts for reproducing the experiments in the paper #1

Open nishadsinghi opened 1 year ago

nishadsinghi commented 1 year ago

Hey, guys! Thanks for your work, I really enjoyed reading it. Also, thanks for sharing the code. :) I was wondering if it would be possible for you to share the scripts you used to perform all the experiments in the paper, so I can reproduce the results.

ejkim47 commented 1 year ago

Thanks for your interest in my work.

The current code includes a script for training and evaluating ProbCBM on the CUB dataset. Can you tell me specifically which script for the experiment you want?

nishadsinghi commented 1 year ago

It would be helpful to have the scripts used to generate the synthetic dataset (consisting of MNIST images) and the ones used to generate figures 6, 8, 10, 11. :)

nishadsinghi commented 1 year ago

Additionally, if you already have scripts for generating figures 4, 5, 7, 9, 12, I would very much appreciate it if you could share them as well! :)

ejkim47 commented 1 year ago

The code for generating all things you're talking about is not released yet. We will provide additional code and script as soon as it is cleaned up.

Thanks:)