apache/beam

[Task]: Create a helper function fill_in_missing for TFT Criteo tests

開放

#24,902 建立於 2023年1月5日

 (5 則留言) (0 個反應) (1 位負責人)Java (4,097 個分叉)batch import
P3good first issuepythonrun-inferencetask

倉庫指標

星標
 (7,313 顆星)
PR 合併指標
 (PR 指標待抓取)

描述

What needs to happen?

This function should accept a rank 2 SparseTensor, and a default value, and return a rank 1 Tensor. It will assume the input is from a VarLenFeature and has dimensions [batch_size, 0] or [batch_size, 1] depending on the max size of the feature over the batch. It's assumed each feature has 0 or 1 values (0 for missing, 1 for present).

It will emit a Tensor which is constructed using the code

  feature = tf.sparse_to_dense(
      feature.indices, [feature.dense_shape[0], 1], feature.values,
      default_value=-1)
  feature = tf.squeeze(feature, axis=1)

Issue Priority

Priority: 3 (nice-to-have improvement)

Issue Components

  • Component: Python SDK
  • Component: Java SDK
  • Component: Go SDK
  • Component: Typescript SDK
  • Component: IO connector
  • Component: Beam examples
  • Component: Beam playground
  • Component: Beam katas
  • Component: Website
  • Component: Spark Runner
  • Component: Flink Runner
  • Component: Samza Runner
  • Component: Twister2 Runner
  • Component: Hazelcast Jet Runner
  • Component: Google Cloud Dataflow Runner

貢獻者指南