folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Select variables to be used as labels in _Nd_sampler (#47)

Browse Source 
* select labels variables
This commit is contained in:
Francesco Andreuzzi
2022-12-06 11:35:37 +01:00
committed by GitHub
parent e974801df2
commit 20dd11d586
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
pina/span.py
View File

@@ -84,12 +84,12 @@ class Span(Location):
def _Nd_sampler(n, mode, variables): def _Nd_sampler(n, mode, variables):
""" Sample ll the variables together """ """ Sample ll the variables together """
bounds = torch.tensor( pairs = [(k, v) for k, v in self.range_.items() if k in variables]
[v for k, v in self.range_.items() if k in variables] keys, values = map(list, zip(*pairs))
) bounds = torch.tensor(values)
result = self._sample_range(n, mode, bounds) result = self._sample_range(n, mode, bounds)
result = result.as_subclass(LabelTensor) result = result.as_subclass(LabelTensor)
result.labels = list(self.range_.keys()) result.labels = keys
for variable in variables: for variable in variables:
if variable in self.fixed_.keys(): if variable in self.fixed_.keys():