modified: __init__.py

modified: dsm_torch.py
autonlab · Mar 1, 2022 · 177f9af · 177f9af
1 parent 33dc15c
commit 177f9af
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/auton_survival/models/dsm/__init__.py b/auton_survival/models/dsm/__init__.py
@@ -60,6 +60,9 @@
 Deep Recurrent Survival Machines
 --------------------------------
 
+<img width=100% src=https://ndownloader.figshare.com/files/28329918>
+
+
 **Deep Recurrent Survival Machines (DRSM)** builds on the original **DSM**
 model and allows for learning of representations of the input covariates using
 **Recurrent Neural Networks** like **LSTMs, GRUs**. Deep Recurrent Survival

diff --git a/auton_survival/models/dsm/dsm_torch.py b/auton_survival/models/dsm/dsm_torch.py
@@ -91,7 +91,6 @@ def create_representation(inputdim, layers, activation, bias=False):
   return nn.Sequential(*modules)
 
 
-
 class DeepSurvivalMachinesTorch(torch.nn.Module):
   """A Torch implementation of Deep Survival Machines model.
Original file line number	Diff line number	Diff line change
Expand Up		@@ -91,7 +91,6 @@ def create_representation(inputdim, layers, activation, bias=False):
		return nn.Sequential(*modules)



		class DeepSurvivalMachinesTorch(torch.nn.Module):
		"""A Torch implementation of Deep Survival Machines model.

Expand Down