search

liupei101 / TFDeepSurv

COX Proportional risk model and survival analysis implemented by tensorflow.
MIT License
99 stars 27 forks source link

Add sklearn to required packages. #11

Closed taketakeseijin closed 3 years ago

taketakeseijin commented 3 years ago

tfdeepsurv needs sklearn in tfdeepsurv.datasets, but it's not listed in setup.py as a required package.