diff --git a/python/fate/ml/intersection/raw_intersection.py b/python/fate/ml/intersection/raw_intersection.py index b294ddea54..a037470871 100644 --- a/python/fate/ml/intersection/raw_intersection.py +++ b/python/fate/ml/intersection/raw_intersection.py @@ -33,6 +33,10 @@ def fit(self, ctx: Context, train_data, validate_data=None): for intersect_index in intersect_indexes: intersect_data = intersect_data.loc(intersect_index) + intersect_count = intersect_data.count() + ctx.hosts.put("intersect_count", intersect_count) + + logger.info(f"intersect count={intersect_count}") return intersect_data @@ -46,4 +50,6 @@ def fit(self, ctx: Context, train_data, validate_data=None): # ctx.guest.put("intersect_index", intersect_data.index.tolist()) ctx.guest.put("intersect_index", intersect_data.index.values) + intersect_count = ctx.guest.get("intersect_count") + logger.info(f"intersect count={intersect_count}") return intersect_data