ensure ordinal encoder returns concrete type

FluxML · Sep 26, 2024 · 87c8f66 · 87c8f66
1 parent 804c0c5
commit 87c8f66
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)