當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.feature_column.sequence_categorical_column_with_hash_bucket用法及代碼示例


通過散列設置 id 的分類術語序列。

用法

tf.feature_column.sequence_categorical_column_with_hash_bucket(
    key, hash_bucket_size, dtype=tf.dtypes.string
)

參數

  • key 標識輸入特征的唯一字符串。
  • hash_bucket_size 一個 int > 1。桶的數量。
  • dtype 特征的類型。僅支持字符串和整數類型。

返回

  • 一個SequenceCategoricalColumn。

拋出

  • ValueError hash_bucket_size 不大於 1。
  • ValueError dtype 既不是字符串也不是整數。

將此傳遞給 embedding_column 或 indicator_column 以將序列分類數據轉換為密集表示，以輸入到序列 NN，例如 RNN。

例子：

tokens = sequence_categorical_column_with_hash_bucket(
    'tokens', hash_bucket_size=1000)
tokens_embedding = embedding_column(tokens, dimension=10)
columns = [tokens_embedding]

features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
sequence_feature_layer = SequenceFeatures(columns)
sequence_input, sequence_length = sequence_feature_layer(features)
sequence_length_mask = tf.sequence_mask(sequence_length)

rnn_cell = tf.keras.layers.SimpleRNNCell(hidden_size)
rnn_layer = tf.keras.layers.RNN(rnn_cell)
outputs, state = rnn_layer(sequence_input, mask=sequence_length_mask)

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.feature_column.sequence_categorical_column_with_hash_bucket。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。