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

將權重值應用於 CategoricalColumn 。

用法

tf.feature_column.weighted_categorical_column(
    categorical_column, weight_feature_key, dtype=tf.dtypes.float32
)

當您的每個稀疏輸入都具有 ID 和值時，請使用此選項。例如，如果您將文本文檔表示為詞頻的集合，則可以提供 2 個並行稀疏輸入特征(下麵的'terms' 和'frequencies')。

例子：

輸入 tf.Example 對象：

[
  features {
    feature {
      key:"terms"
      value {bytes_list {value:"very" value:"model"} }
    }
    feature {
      key:"frequencies"
      value {float_list {value:0.3 value:0.1} }
    }
  },
  features {
    feature {
      key:"terms"
      value {bytes_list {value:"when" value:"course" value:"human"} }
    }
    feature {
      key:"frequencies"
      value {float_list {value:0.4 value:0.1 value:0.2} }
    }
  }
]

categorical_column = categorical_column_with_hash_bucket(
    column_name='terms', hash_bucket_size=1000)
weighted_column = weighted_categorical_column(
    categorical_column=categorical_column, weight_feature_key='frequencies')
columns = [weighted_column, ...]
features = tf.io.parse_example(..., features=make_parse_example_spec(columns))
linear_prediction, _, _ = linear_model(features, columns)

這假設輸入字典包含鍵 'terms' 的 SparseTensor 和鍵 'frequencies' 的 SparseTensor。這 2 個張量必須具有相同的索引和密集形狀。