Module: tff.analytics.data_processing

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.