Supported frameworks (Watson Machine Learning)
You can use popular tools, libraries, and frameworks to train and deploy machine learning models and functions using IBM Watson Machine Learning. This topic lists supported versions and features.
View Using Watson Machine Learning in a notebook for links to sample notebooks that demonstrate training and deploying models and functions using supported frameworks.
Supported machine learning frameworks
- Support for Python version 3.6 is deprecated in favor of Python version 3.7.
- If a framework is marked as deprecated then support for the framework will be removed in a future release. For details on migrating a model or function, see Upgrading from a deprecated framework.
- Discontinued frameworks include:
- Caffe 1.0
- Pytorch 1.0
- Spark 2.3
- The way you can work with deprecated frameworks depends on whether you upgraded Cloud Pak for Data from a previous version or did a new installation of Cloud Pak for Data 3.5. The difference is:
- For new installations of Cloud Pak for Data 3.5, only non-deprecated frameworks are supported.
- If you upgraded from a previous release of Cloud Pak for Data, you can use deprecated frameworks.
- AutoAI models trained in previous versions of Cloud Pak for Data are deprecated as they are based on Python 3.6. You must retrain the experiments.
- If you are creating a CoreML deployment for a logistic regression model using Scikit-learn 0.23 with Python 3.7, you must explicitly specify a value to override the default Scikit-learn package with Scikit-learn 0.23 during model training. See this article for details of specifying a package version when you are training a Scikit-learn pipeline.
- If you are using Pytorch 1.3 with Python 3.7 to train a model, you must explicitly set
keep_initializers_as_inputs=Truewhen exporting your model.
- Scikit-learn 0.22 , Xgboost 0.82, Xgboost 0.90 CoreML deployments with Python 3.6 will not work on ROKS vpcgen2 with Portworx storage cluster.
- In order for a Scikit-learn or Xgboost model to be converted to CoreML format, it must contain only supported transformer algorithms. Refer to this documentation for lists of supported CoreML transformers for Scikit-learn and Xgboost frameworks: