diff --git a/auton_survival/models/dcm/__init__.py b/auton_survival/models/dcm/__init__.py index 7d0d9d3..72b44c9 100644 --- a/auton_survival/models/dcm/__init__.py +++ b/auton_survival/models/dcm/__init__.py @@ -87,8 +87,8 @@ class DeepCoxMixtures: """ - def __init__(self, k=3, layers=None, gamma=1, - smoothing_factor=1e-2, use_activation=False): + def __init__(self, k=3, layers=None, gamma=10, + smoothing_factor=1e-4, use_activation=False): self.k = k self.layers = layers