Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add general coupling transformation #23

Merged
merged 6 commits into from
Jul 13, 2023
Merged

Add general coupling transformation #23

merged 6 commits into from
Jul 13, 2023

Conversation

simonschnake
Copy link
Contributor

@simonschnake simonschnake commented Jul 11, 2023 •
edited by francois-rozet
Loading

Adds

  • Transform: CouplingTransform
  • TransformModule: MaskedCouplingTransform
  • FlowModule: CouplingFlow

Details discussed in the discussion #22

Simon Schnake and others added 3 commits July 11, 2023 11:38
Initial CouplingFlow
d7d6fd1
@francois-rozet
Merge branch 'master' into CouplingFlow
94b8c92
@francois-rozet
Changes
e2fcd5f 
* Rename to GeneralCouplingTransform to match Dinh et al. (2014)
* Use new unpack helper
* Use checkered coupling masks by default
* Use more concise notations in CouplingTransform
* Fix GCF's __init__ and extra_repr
@francois-rozet
Copy link
Member

francois-rozet commented Jul 13, 2023 •
edited
Loading

Hello @simonschnake, I have made a few changes to the PR. It is a major feature addition so it required some care, notably for the tests. I have settled for the name GeneralCouplingTransform as written by Dinh et al. (2014). I also fixed a few issues with the CouplingFlow (renamed GCF for consistency). Most of the architecture and docstrings remains the same, though, so good job! I will do some more tests, but I think it is ready to merge otherwise 🔥

@francois-rozet francois-rozet changed the title CouplingFlow Add general coupling transformation Jul 13, 2023
@francois-rozet francois-rozet merged commit 85a0ff9 into probabilists:master Jul 13, 2023
@simonschnake simonschnake deleted the CouplingFlow branch July 14, 2023 11:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@simonschnake @francois-rozet