Skip to content

Commit d08a4f3

Browse files
committed
Fix division in generate_label_transfer_dict to prevent division by zero by adding a small constant to the denominator.
1 parent a05205a commit d08a4f3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

spateo/alignment/methods/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -432,7 +432,7 @@ def generate_label_transfer_dict(
432432
norm_label_transfer_dict = dict()
433433
for c1 in cat1:
434434
norm_c = np.array([label_transfer_dict[c1][c2] for c2 in cat2]).sum()
435-
norm_label_transfer_dict[c1] = {c2: label_transfer_dict[c1][c2] / norm_c for c2 in cat2}
435+
norm_label_transfer_dict[c1] = {c2: label_transfer_dict[c1][c2] / (norm_c + 1e-8) for c2 in cat2}
436436

437437
return norm_label_transfer_dict
438438

0 commit comments

Comments
 (0)