Skip to content

Commit

Permalink
Made clear only highly_variable features are used
Browse files Browse the repository at this point in the history
  • Loading branch information
gjhuizing committed Nov 7, 2023
1 parent 1c17ac7 commit 4f1775b
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions mowgli/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,9 @@ class MowgliModel:
Args:
latent_dim (int, optional):
The latent dimension of the model. Defaults to 15.
highly_variable (bool, optional):
Whether to use highly variable features. Defaults to True.
For now, only True is supported.
use_mod_weight (bool, optional):
Whether to use a different weight for each modality and each
cell. If `True`, the weights are expected in the `mod_weight`
Expand Down Expand Up @@ -47,6 +50,7 @@ class MowgliModel:
def __init__(
self,
latent_dim: int = 50,
highly_variable: bool = True,
use_mod_weight: bool = False,
h_regularization: float = {
"rna": 1e-2,
Expand All @@ -65,6 +69,8 @@ def __init__(
assert latent_dim > 0
assert w_regularization > 0
assert eps > 0
assert highly_variable is True
# TODO: Actually implement the use of highly_variable

if isinstance(h_regularization, dict):
for mod in h_regularization:
Expand Down

0 comments on commit 4f1775b

Please sign in to comment.