Merge pull request #11 from Jeanselme/patch-1

Update initialisation
autonlab · Nov 9, 2020 · 3e3ed66 · 3e3ed66
2 parents cb84eb8 + ea15f7f
commit 3e3ed66
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dsm/dsm_api.py b/dsm/dsm_api.py
@@ -252,7 +252,8 @@ class DeepRecurrentSurvivalMachines(DSMBase):
   def __init__(self, k=3, layers=None, hidden=None, 
                distribution='Weibull', temp=1000., discount=1.0, typ='LSTM'):
     super(DeepRecurrentSurvivalMachines, self).__init__(k=k, layers=layers,
-                                                        distribution=distribution)
+                                                        distribution=distribution,
+                                                        temp=temp, discount=discount)
     self.hidden = hidden
     self.typ = typ
   def _gen_torch_model(self, inputdim, optimizer):