Skip to content

Commit

Permalink
modified: estimators.py
Browse files Browse the repository at this point in the history
	modified:   phenotyping.py
  • Loading branch information
chiragnagpal committed Mar 29, 2022
1 parent c65f315 commit cefc51c
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 13 deletions.
10 changes: 5 additions & 5 deletions auton_survival/estimators.py
Original file line number Diff line number Diff line change
Expand Up @@ -600,11 +600,11 @@ class SurvivalModel:
A string that determines the choice of the surival analysis model.
Survival model choices include:
- 'dsm' : Deep Survival Machines [3] model
- 'dcph' : Deep Cox Proportional Hazards [2] model
- 'dcm' : Deep Cox Mixtures [4] model
- 'rsf' : Random Survival Forests [1] model
- 'cph' : Cox Proportional Hazards [2] model
- `dsm` : Deep Survival Machines [3] model
- `dcph` : Deep Cox Proportional Hazards [2] model
- `dcm` : Deep Cox Mixtures [4] model
- `rsf` : Random Survival Forests [1] model
- `cph` : Cox Proportional Hazards [2] model
random_seed: int
Controls the reproducibility of called functions.
Expand Down
16 changes: 8 additions & 8 deletions auton_survival/phenotyping.py
Original file line number Diff line number Diff line change
Expand Up @@ -221,17 +221,17 @@ class ClusteringPhenotyper(Phenotyper):
The clustering method applied for phenotyping.
Options include:
- 'kmeans': K-Means Clustering
- 'dbscan': Density-Based Spatial Clustering of Applications with Noise (DBSCAN)
- 'gmm': Gaussian Mixture
- 'hierarchical': Agglomerative Clustering
- `kmeans`: K-Means Clustering
- `dbscan`: Density-Based Spatial Clustering of Applications with Noise (DBSCAN)
- `gmm`: Gaussian Mixture
- `hierarchical`: Agglomerative Clustering
dim_red_method: str, default=None
The dimensionality reductions method applied.
Options include:
- 'pca' : Principal Component Analysis
- 'kpca' : Kernel Principal Component Analysis
- 'nnmf' : Non-Negative Matrix Factorization
- `pca` : Principal Component Analysis
- `kpca` : Kernel Principal Component Analysis
- `nnmf` : Non-Negative Matrix Factorization
- None : dimensionality reduction is not applied.
random_seed : int, default=0
Controls the randomness and reproducibility of called functions
Expand Down Expand Up @@ -435,7 +435,7 @@ def __init__(self,

self.cf_method = cf_method
self.phenotyping_method = phenotyping_method

if cf_method_hyperparams is None:
cf_method_hyperparams = {}
if phenotyper_hyperparams is None:
Expand Down

0 comments on commit cefc51c

Please sign in to comment.