google-research / discs

DISCS: The code base for the Benchmark for Discrete Sampling
Apache License 2.0
43 stars 7 forks source link

Refactor experiment #40

Closed KatayoonGoshvadi closed 1 year ago

KatayoonGoshvadi commented 1 year ago

@Hanjun-Dai , Hi Hanjun, In this PR, I mainly refactored the experiment class. We now have a base Experiment class. Dependent on if we are running CO or not, the chain computation happens with the corresponding child class. In case of CO, CO_Experiment is run and in other cases, Sampling_Experiment is run. I also refactored the evaluator class. For adding any new form of evaluation, a child class could be created extending the abstract parent class and the evaluate function need to be overriden. Thanks and I look forward to your comments!