| e {cito} | R Documentation |
Embeddings
Description
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 |
[Package cito version 1.1 Index]