IPKM/mdevaluate
6
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Selector now checks for to many dimensions.

Browse Source
This commit is contained in:
Sebastian Kloth 2024-01-09 13:46:46 +01:00
parent 0870a94b44
commit a559f72221
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/mdevaluate/correlation.py
View File

@ -100,16 +100,25 @@ def shifted_correlation(
else:
index = selector(frames[start_frame])
if len(index) == 0:
return np.zeros(len(idx))
return np.zeros(len(shifted_idx))
if isinstance(index[0], int) or isinstance(index[0], bool):
elif isinstance(index[0], int) or isinstance(index[0], bool):
return get_correlation(frames, start_frame, index, shifted_idx)
else:
correlations = []
for ind in index:
if len(ind) == 0:
correlations.append(np.zeros(len(shifted_idx)))
elif isinstance(ind[0], int) or isinstance(ind[0], bool):
correlations.append(
get_correlation(frames, start_frame, ind, shifted_idx)
)
else:
raise ValueError(
"selector has more than two dimensions or does not "
"contain int or bool types"
)
return correlations
if 1 - skip < window: