Xtra-Computing / thundergbm

ThunderGBM: Fast GBDTs and Random Forests on GPUs
Apache License 2.0
689 stars 85 forks source link

how to save and load model in python? #12

Closed CupidJay closed 5 years ago

CupidJay commented 5 years ago

I tried to use joblib and pickle to save a trained model in python, but i got a 'TypeError: can't pickle module objects' error. Hence, I wonder how can i save and load thundergbm model in python. Thanks for your help!

CupidJay commented 5 years ago

I have found model_in, model_out parameters in documents. Sorry to trouble you and I'll close this issue.