secure-xgboost icon indicating copy to clipboard operation
secure-xgboost copied to clipboard

Enable `load_model()` from result of `save_raw()`

Open chester-leung opened this issue 3 years ago • 0 comments

In vanilla XGBoost, the user can call load_model() on the output of save_raw() , which is an in-memory buffer – a serialized model.

In Secure XGBoost, we don’t yet support this functionality when we have a remote client. We need to add support for this by transferring the buffer returned from save_raw() over gRPC to the server, where the buffer (instead of a filename string) can be passed to load_model(). To do so, we'll have to modify the gRPC request message for load_model() to optionally contain the serialized buffer (as well as optionally contain a path to a saved model), and pass this serialized buffer to the server side call for load_model().

chester-leung avatar May 06 '21 23:05 chester-leung