modified: __init__.py

modified: dcm_api.py modified: dcm_torch.py
autonlab · Mar 15, 2022 · 64c3de0 · 64c3de0
1 parent 2623bf8
commit 64c3de0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/auton_survival/models/dcm/__init__.py b/auton_survival/models/dcm/__init__.py
@@ -81,7 +81,7 @@ class DeepCoxMixtures:
       hidden layer.
   Example
   -------
-  >>> from dsm.contrib import DeepCoxMixtures
+  >>> from auton_survival.models.dcm import DeepCoxMixtures
   >>> model = DeepCoxMixtures()
   >>> model.fit(x, t, e)
 

diff --git a/auton_survival/models/dcm/dcm_api.py b/auton_survival/models/dcm/dcm_api.py
@@ -6,7 +6,7 @@
 
 
 class DeepCoxMixtures:
-  """A Deep Cox Mixture model.
+  """A Deep Cox Mixtures model.
 
   This is the main interface to a Deep Cox Mixture model.
   A model is instantiated with approporiate set of hyperparameters and
@@ -195,4 +195,4 @@ def predict_latent_z(self, x):
                       "model using the `fit` method on some training data " +
                       "before calling `predict_latent_z`.")
 
- 
+
diff --git a/auton_survival/models/dcm/dcm_torch.py b/auton_survival/models/dcm/dcm_torch.py
@@ -3,7 +3,7 @@
 
 import numpy as np
 
-from dsm.dsm_torch import create_representation
+from auton_survival.models.dsm.dsm_torch import create_representation
 
 class DeepCoxMixturesTorch(nn.Module):
   """PyTorch model definition of the Deep Cox Mixture Survival Model.