Skip to content

Commit 9892d59

Browse files
committed
Normalizing deletion model
1 parent 9a84e12 commit 9892d59

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

neat/models/models.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,9 @@ class DeletionModel(VariantModel):
9090
def __init__(self,
9191
deletion_len_model: dict[int: float, ...],
9292
rng: Generator = None):
93-
self.deletion_len_model = deletion_len_model
93+
max_value = max(deletion_len_model.values())
94+
# normalize the values
95+
self.deletion_len_model = {key: val/max_value for key, val in deletion_len_model.items()}
9496
self.rng = rng
9597

9698
def get_deletion_length(self, size: int = None) -> int | list[int, ...]:

0 commit comments

Comments
 (0)