Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/entity-embeddings' into double-t…
Browse files Browse the repository at this point in the history
…rouble
  • Loading branch information
ablaom committed Sep 26, 2024
2 parents 09b17da + c57870b commit 804c0c5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/encoders.jl
Original file line number Diff line number Diff line change
@@ -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)

0 comments on commit 804c0c5

Please sign in to comment.