unwrap after recoding

FluxML · Nov 1, 2024 · 268afad · 268afad
1 parent 3315da4
commit 268afad
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(col, level2scalar...)
+            new_col = unwrap.(recode(col, level2scalar...))
             push!(new_feats, new_col)
         else
             push!(new_feats, col)