survml Goto Github PK

.github

auton-survival

Auton Survival - an open source package for Regression, Counterfactual Estimation, Evaluation and Phenotyping with Censored Time-to-Events

survml-about

Organization files

survml-deepsurv

PyTorch implementation of DeepSurv, by Katzman, J. L., Shaham, U., Cloninger, A., Bates, J., Jiang, T., & Kluger, Y. (2018). DeepSurv: personalized treatment recommender system using a Cox proportional hazards deep neural network. BMC medical research methodology, 18(1), 1-12.

survml-deepweisurv

PyTorch implementation of DeepWeiSurv, by Bennis, A., Mouysset, S., & Serrurier, M. (2020, May). Estimation of conditional mixture Weibull distribution with right censored data using neural network for time-to-event analysis. In Pacific-Asia Conference on Knowledge Discovery and Data Mining (pp. 687-698). Springer, Cham.

survml-lstm-rul-prediction

PyTorch implementation of remaining useful life prediction with long-short term memories (LSTM), performing on NASA C-MAPSS data sets. Partially inspired by Zheng, S., Ristovski, K., Farahat, A., & Gupta, C. (2017, June). Long short-term memory network for remaining useful life estimation.

survml-transformer-rul-prediction

Transformer implementation with PyTorch for remaining useful life prediction on turbofan engine with NASA CMAPSS data set. Inspired by Mo, Y., Wu, Q., Li, X., & Huang, B. (2021). Remaining useful life estimation via transformer encoder enhanced by a gated convolutional unit. Journal of Intelligent Manufacturing, 1-10.