Skip to content

Commit 08842c0

Browse files
committed
Remove unused cat code
1 parent 0c27de5 commit 08842c0

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

src/simdec/decomposition.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -109,11 +109,9 @@ def decomposition(
109109
var_names = inputs.columns
110110

111111
cat_cols = inputs.select_dtypes(exclude=["number"])
112-
cat_cols_idx = []
113112
for cat_col in cat_cols:
114113
codes, cat_states_ = pd.factorize(inputs[cat_col])
115114
inputs[cat_col] = codes
116-
cat_cols_idx.append(inputs.columns.get_loc(cat_col))
117115

118116
inputs = inputs.to_numpy()
119117
output = output.to_numpy()
@@ -136,7 +134,6 @@ def decomposition(
136134
states = 3 if n_var_dec < 3 else 2
137135
states = [states] * n_var_dec
138136

139-
# categorical for a given variable
140137
for i in range(n_var_dec):
141138
n_unique = np.unique(inputs[:, i]).size
142139
states[i] = n_unique if n_unique <= 5 else states[i]

0 commit comments

Comments
 (0)