autonlab / auton-survival

Auton Survival - an open source package for Regression, Counterfactual Estimation, Evaluation and Phenotyping with Censored Time-to-Events
http://autonlab.github.io/auton-survival
MIT License
315 stars 74 forks source link

saving models? #113

Closed zapaishchykova closed 1 year ago

zapaishchykova commented 1 year ago

Hi, is there any way to save and restore the model?

chiragnagpal commented 1 year ago

try dill import dill as pkl dill.dump(model, open("saved_model.pkl", "wb))