Add scikit-learn joblib documentation

This commit is contained in:
allegroai 2019-07-31 23:53:52 +03:00
parent ee3fa31811
commit bcaf4ce6c8

View File

@ -381,7 +381,12 @@ Yes! Check our [TRAINS Jupyter Plugin](https://github.com/allegroai/trains-jupyt
Yes! `scikit-learn` is supported. Everything you do is logged.
**NOTE**: Models are not automatically logged because in most cases, scikit-learn will simply pickle the object to files so there is no underlying frame we can connect to.
Models are automatically logged when stored using *joblib*.
# from sklearn.externals import joblib
import joblib
joblib.dump(model, 'model.pkl')
loaded_model = joblib.load('model.pkl')
## TRAINS API