From c57870ba5139b591c3bec0303acf8ad815d7db96 Mon Sep 17 00:00:00 2001 From: Essam Date: Wed, 25 Sep 2024 20:11:33 -0500 Subject: [PATCH] Fix ordinal encoding ransform type --- src/encoders.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/encoders.jl b/src/encoders.jl index a7eba0b..4d00d2d 100644 --- 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 = recode(unwrap.(col), level2scalar...) push!(new_feats, new_col) else push!(new_feats, col)