, . , (), . , . . , var-length ( ) (, 256- ).
_CategoricalColumn.
cat_column_with_vocab = tf.feature_column.categorical_column_with_vocabulary_list(
key='my-text',
vocabulary_list=vocab_list)
, , categorical_column_with_vocabulary_file.
, ( ) .
embedding_initializer = None
if has_pretrained_embedding:
embedding_initializer=tf.contrib.framework.load_embedding_initializer(
ckpt_path=xxxx)
else:
embedding_initializer=tf.random_uniform_initializer(-1.0, 1.0)
embed_column = embedding_column(
categorical_column=cat_column_with_vocab,
dimension=256, ## this is your pre-trained embedding dimension
initializer=embedding_initializer,
trainable=False)
, price:
price_column = tf.feature_column.numeric_column('price')
columns = [embed_column, price_column]
:
features = tf.parse_example(...,
features=make_parse_example_spec(columns))
dense_tensor = tf.feature_column.input_layer(features, columns)
for units in [128, 64, 32]:
dense_tensor = tf.layers.dense(dense_tensor, units, tf.nn.relu)
prediction = tf.layers.dense(dense_tensor, 1)
, tf.parse_example , tf.Example ( protobuf):
features {
feature {
key: "price"
value { float_list {
value: 29.0
}}
}
feature {
key: "my-text"
value { bytes_list {
value: "this"
value: "product"
value: "is"
value: "for sale"
value: "within"
value: "us"
}}
}
}
, : - , - .
["this", "product", "is", "for sale", "within", "us"].