Skip to content

Commit

Permalink
add delta function testing prior (#183)
Browse files Browse the repository at this point in the history
  • Loading branch information
deepchatterjeeligo authored Dec 4, 2024
1 parent 5b93414 commit bec210b
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions amplfi/train/priors.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,3 +54,29 @@ def cbc_prior() -> ParameterSampler:
torch.as_tensor(0, dtype=torch.float32),
),
)


def cbc_testing_delta_function_prior() -> ParameterSampler:
return ParameterSampler(
chirp_mass=distributions.DeltaFunction(
torch.as_tensor(55, dtype=torch.float32),
),
mass_ratio=distributions.DeltaFunction(
torch.as_tensor(0.9, dtype=torch.float32),
),
distance=distributions.DeltaFunction(
torch.as_tensor(1000, dtype=torch.float32),
),
inclination=distributions.DeltaFunction(
torch.as_tensor(torch.pi / 6, dtype=torch.float32),
),
phic=distributions.DeltaFunction(
torch.as_tensor(torch.pi, dtype=torch.float32),
),
chi1=distributions.DeltaFunction(
torch.as_tensor(0, dtype=torch.float32),
),
chi2=distributions.DeltaFunction(
torch.as_tensor(0, dtype=torch.float32),
),
)

0 comments on commit bec210b

Please sign in to comment.