diff --git a/python/federatedml/statistic/statics.py b/python/federatedml/statistic/statics.py index fab4d4efcd..967604767a 100755 --- a/python/federatedml/statistic/statics.py +++ b/python/federatedml/statistic/statics.py @@ -587,6 +587,9 @@ def merge_count_dict(x, y): return x data_mode_summary = data_reduce_by_col_val.mapPartitions(func).reduce(merge_count_dict) + diff = set(data_overview.get_header(data)).difference(set(data_mode_summary.keys())) + if diff: + LOGGER.warning(f"Following columns do not have mode: {diff}.") return data_mode_summary def get_mode(self, col=None, multi_mode='random'):