Update dcm_api.py

autonlab · Mar 8, 2022 · 45a5114 · 45a5114
1 parent f4a2729
commit 45a5114
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/auton_survival/models/dcm/dcm_api.py b/auton_survival/models/dcm/dcm_api.py
@@ -35,8 +35,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
@@ -195,4 +195,4 @@ def predict_latent_z(self, x):
                       "model using the `fit` method on some training data " +
                       "before calling `predict_latent_z`.")
 
-
+