select random mode when counts equal

Signed-off-by: Yu Wu <[email protected]>
FederatedAI · Nov 22, 2023 · c21a8ba · c21a8ba
1 parent ec38c52
commit c21a8ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/federatedml/statistic/statics.py b/python/federatedml/statistic/statics.py
@@ -566,7 +566,7 @@ def __aggregate_count_per_val_in_col(kv_iterator, header):
                     if col_dict[col_name]['max_count'] < v:
                         col_dict[col_name] = {'max_count': v, 'max_val': [k[1]]}
                     elif col_dict[col_name]['max_count'] == v:
-                        col_dict[col_name]['max_val'].append(k[1])
+                        col_dict[col_name]['max_val'] = random.choice([k[1], col_dict[col_name]['max_val']])
             return col_dict
 
         func = functools.partial(__aggregate_count_per_val_in_col, header=data.schema.get('header'))