diff --git a/src/encoders.jl b/src/encoders.jl index 5085ce0..f396142 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(unwrap.(col), level2scalar...)...] + new_col = recode(unwrap.(col), level2scalar...) push!(new_feats, new_col) else push!(new_feats, col)