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.