remove v -> [v...] from output of encoder

FluxML · Sep 29, 2024 · bc805e4 · bc805e4
1 parent b2bafd2
commit bc805e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/encoders.jl b/src/encoders.jl
@@ -67,7 +67,7 @@ function ordinal_encoder_transform(X, mapping_matrix)
             test_levels = levels(col)
             check_unkown_levels(train_levels, test_levels)
             level2scalar = mapping_matrix[ind]
-            new_col = [recode(unwrap.(col), level2scalar...)...]
+            new_col = recode(unwrap.(col), level2scalar...)
             push!(new_feats, new_col)
         else
             push!(new_feats, col)