View source on GitHub
|
A set of utility functions for data processing.
Functions
get_all_elements(...): Gets all the elements from the input dataset.
get_capped_elements(...): Gets the first max_user_contribution elements from the input dataset.
get_capped_elements_with_counts(...): Gets the capped elements with counts from the input dataset.
get_top_elements(...): Gets top unique elements from the input dataset.
get_top_elements_with_counts(...): Gets top unique elements from the input dataset.
get_top_multi_elements(...): Gets the top unique word multiset from the input dataset.
get_unique_elements(...): Gets the unique elements from the input dataset.
get_unique_elements_with_counts(...): Gets unique elements and their counts from the input dataset.
to_stacked_tensor(...): Encodes the `tf.data.Dataset as stacked tensors.