modified: __init__.py

autonlab · Mar 29, 2022 · 12a7a4e · 12a7a4e
1 parent 78beb2b
commit 12a7a4e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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