modified: dcm_torch.py

dsm/contrib/dcm_torch.py

@@ -13,7 +13,7 @@
 
 class DeepCoxMixturesTorch(nn.Module):
   """PyTorch model definition of the Deep Cox Mixture Survival Model.
-  
+
   The Cox Mixture involves the assumption that the survival function
   of the individual to be a mixture of K Cox Models. Conditioned on each
   subgroup Z=k; the PH assumptions are assumed to hold and the baseline