Can be used for categorical variables, a more efficient alternative to one-hot encoding
Usage
e(dim = 1L, weights = NULL, train = TRUE, lambda = 0, alpha = 1)
Arguments
- dim
integer, embedding dimension
- weights
matrix, to use custom embedding matrices
- train
logical, should the embeddings be trained or not
- lambda
regularization strength on the embeddings
- alpha
mix between L1 and L2 regularization