search

arogozhnikov / hep_ml

Machine Learning for High Energy Physics.
https://arogozhnikov.github.io/hep_ml/
Other
176 stars 64 forks source link

Persistify GBReweighter instance #64

Closed acampove closed 3 years ago

acampove commented 3 years ago

Hello Alex,

I am using the GBReweighter class:

https://arogozhnikov.github.io/hep_ml/reweight.html?highlight=gbreweighter#hep_ml.reweight.GBReweighter.predict_weights

and I would like to know if there is a native way to persistify an instance of this class so that it can be used to predict_weights afterwards. I cannot find anything native in the documentation and I was thinking that maybe something like pickle might work.

Cheers.

arogozhnikov commented 3 years ago

Hi @acampove , yes, pickle is the right way to store estimators from hep_ml including GBReweighter.