Confusion regarding joblib.dump()

Confusion regarding joblib.dump()

Problem Description:

One way to save sklearn models is to use joblib.dump(model,filename). I have a confusion regarding the filename argument. One way to run this function is through :


This saves the model successfully and also the model is loaded correctly using the:


Another way is to use :


With no ".joblib" extension this time. This also runs successfully and the model is loaded correctly using the:


What confuses me is the file extension in the filename, Is there a certain file extension that I should use for saving the model? Or it is not necessary to use a file extension as I did above? If it is not necessary, then why?

Solution – 1

There is no file extension that "must" be used to serialize a model. You can specify the compression method by using one of the supported filename extensions (.z, .gz, .bz2, .xz or .lzma). By default joblib will use zlib to serialize objects.

Therefore you can use any file extension. However it is a good practice to use the library as the extension in order to know how to load it.

I name my serialized model model.pickle when I am using pickle library and model.joblib when I am using joblib.

Rate this post
We use cookies in order to give you the best possible experience on our website. By continuing to use this site, you agree to our use of cookies.