wassname/pytorch-ts
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-ts.git synced 2026-07-07 18:53:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix name clash

Browse Source
This commit is contained in:
Dr. Kashif Rasul
2020-05-15 13:57:16 +02:00
parent cd5b061b9b
commit 2f65617be0
1 changed files with 10 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pts/modules/distribution_output.py
+10 -14
View File
@@ -191,16 +191,15 @@ class StudentTMixtureOutput(DistributionOutput):
def distribution(
self, distr_args, scale: Optional[torch.Tensor] = None
) -> Distribution:
mix_logits, df, loc, scale = distr_args
mix_logits, df, loc, dist_scale = distr_args
comp_distr = StudentT(df, loc, scale)
distr = MixtureSameFamily(
Categorical(logits=mix_logits), StudentT(df, loc, dist_scale)
)
if scale is None:
return MixtureSameFamily(Categorical(logits=mix_logits), comp_distr)
return distr
else:
scaled_comp_distr = TransformedDistribution(
comp_distr, [AffineTransform(loc=0, scale=scale)]
)
return MixtureSameFamily(Categorical(logits=mix_logits), scaled_comp_distr)
return TransformedDistribution(distr, [AffineTransform(loc=0, scale=scale)])
@property
def event_shape(self) -> Tuple:
@@ -224,16 +223,13 @@ class NormalMixtureOutput(DistributionOutput):
def distribution(
self, distr_args, scale: Optional[torch.Tensor] = None
) -> Distribution:
mix_logits, loc, scale = distr_args
mix_logits, loc, dist_scale = distr_args
comp_distr = Normal(loc, scale)
distr = MixtureSameFamily(Categorical(logits=mix_logits), Normal(loc, dist_scale))
if scale is None:
return MixtureSameFamily(Categorical(logits=mix_logits), comp_distr)
return distr
else:
scaled_comp_distr = TransformedDistribution(
comp_distr, [AffineTransform(loc=0, scale=scale)]
)
return MixtureSameFamily(Categorical(logits=mix_logits), scaled_comp_distr)
return TransformedDistribution(distr, [AffineTransform(loc=0, scale=scale)])
@property
def event_shape(self) -> Tuple: