recursionpharma / gflownet

GFlowNet library specialized for graph & molecular data
MIT License
216 stars 43 forks source link

Adds ActionIndex and Harmonize use of action-masks #129

Closed julienroyd closed 7 months ago

julienroyd commented 8 months ago

Masks were used somewhat ad-hoc here and there. Here we centralise masking in GraphActionCategorical(), specifically:

We also replaced the action-tuples Tuple[int, int, int] action type, row and column by a named-tuple ActionIndex to the code and typing more intuitive and readable.

julienroyd commented 8 months ago

Can reproduce previous results with goal-conditioning and 4 objectives: image