We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0c27de5 commit 08842c0Copy full SHA for 08842c0
src/simdec/decomposition.py
@@ -109,11 +109,9 @@ def decomposition(
109
var_names = inputs.columns
110
111
cat_cols = inputs.select_dtypes(exclude=["number"])
112
- cat_cols_idx = []
113
for cat_col in cat_cols:
114
codes, cat_states_ = pd.factorize(inputs[cat_col])
115
inputs[cat_col] = codes
116
- cat_cols_idx.append(inputs.columns.get_loc(cat_col))
117
118
inputs = inputs.to_numpy()
119
output = output.to_numpy()
@@ -136,7 +134,6 @@ def decomposition(
136
134
states = 3 if n_var_dec < 3 else 2
137
135
states = [states] * n_var_dec
138
139
- # categorical for a given variable
140
for i in range(n_var_dec):
141
n_unique = np.unique(inputs[:, i]).size
142
states[i] = n_unique if n_unique <= 5 else states[i]
0 commit comments