R/columns.R

hub_text_embedding_column

Module to construct a dense representation from a text feature.

Description

This feature column can be used on an input feature whose values are strings of arbitrary size.

Usage

 
hub_text_embedding_column(key, module_spec, trainable = FALSE) 

Arguments

Arguments Description
key A string or feature_column identifying the text feature.
module_spec A string handle or a _ModuleSpec identifying the module.
trainable Whether or not the Module is trainable. FALSE by default, meaning the pre-trained weights are frozen. This is different from the ordinary tf.feature_column.embedding_column(), but that one is intended for training from scratch.