search

vanderschaarlab / autoprognosis

A system for automating the design of predictive modeling pipelines tailored for clinical prognosis.
https://www.autoprognosis.vanderschaar-lab.com/
Apache License 2.0
95 stars 26 forks source link

Saving an autoprognosis model #38

Closed jbwoillard closed 1 year ago

jbwoillard commented 1 year ago

Dear bcebere, how do you save an autoprognosis model file ? It seems tha tpickel does not work for that. Many thanks in advance jb

bcebere commented 1 year ago

Hello @jbwoillard

You can use the save_model/load_model methods from the autoprognosis.utils.serialization module. You can find an example here.

Let me know if this isn't working for you.

jbwoillard commented 1 year ago

Thanks that is perfect