Skip to content

Embeddings  #66

@vitalyli

Description

@vitalyli

Anybody tried to use in-line trained embeddings?
Not all columns are simple numerics. For categoricals or numerics where using bucketization may lead to better
learned representation; using NxM embedding matrix for some features can be helpful.
In Tensorflow it can be done using these helpers:

bucketized_col1 = tf.feature_column.bucketized_column(tf.feature_column.numeric_column(key='col1', shape=[1,], default_value=-2, dtype=tf.int64), boundaries=[1,3,4,5,10,14,18,27,52,61,79])
feature_columns["col1_emb"] = tf.feature_column.embedding_column(bucketized_col1, 5)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions