Skip to content

Commit 1a50f33

Browse files
committed
Fix n_var_dec without auto ordering
1 parent 644ebf5 commit 1a50f33

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

src/simdec/decomposition.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -123,9 +123,13 @@ def decomposition(
123123

124124
# only keep the explained variance corresponding to `dec_limit`
125125
sensitivity_indices = sensitivity_indices[var_order]
126-
n_var_dec = np.where(np.cumsum(sensitivity_indices) < dec_limit)[0].size
127-
n_var_dec = max(1, n_var_dec) # keep at least one variable
128-
n_var_dec = min(5, n_var_dec) # use at most 5 variables
126+
127+
if auto_ordering:
128+
n_var_dec = np.where(np.cumsum(sensitivity_indices) < dec_limit)[0].size
129+
n_var_dec = max(1, n_var_dec) # keep at least one variable
130+
n_var_dec = min(5, n_var_dec) # use at most 5 variables
131+
else:
132+
n_var_dec = inputs.shape[1]
129133

130134
# 2. states formation
131135
if states is None:

0 commit comments

Comments
 (0)