chl8856 / DeepHit

DeepHit: A Deep Learning Approach to Survival Analysis with Competing Risks
181 stars 77 forks source link

Saving the model using joblib etc. #8

Open lawrencepj13 opened 1 year ago

lawrencepj13 commented 1 year ago

Hi there,

I'm trying to save the model to re deploy quickly using test data sets but have had issues using pickle/joblib etc. Is there a method or built in function to save the model? or any recommendations on how to save it?

Many thanks.

Anivader commented 11 months ago

Hello,

I would recommend using this library to run DeepHit using PyTorch - https://github.com/havakv/pycox/blob/master/pycox/models/deephit.py

You can use the "save_model_weights" method to do this.