Supported frameworks
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.
Refer to Machine Learning samples and examples for links to sample notebooks that demonstrate creating batch deployments using the Watson Machine Learning REST API and Watson Machine Learning Python client library.
Important: The tables in this topic document the supported frameworks and software specifications for the current release of Cloud Pak for Data. To see the list of supported frameworks and software specifications for a specific refresh version of Cloud Pak for Data, open the PDF file for "Deploying and managing models and functions" for that refresh version in Documentation for previous 4.0.x refreshes.
Supported machine learning frameworks
| Framework | Versions | Online | Batch |
|---|---|---|---|
| Spark | 3.0 | Yes | Yes Inline payload only |
| PMML | 3.0 to 4.3 | Yes | Programmatic only Inline payload only |
| Hybrid/AutoAI | 0.1 | Yes | Yes |
| SPSS | 18.2 18.1 17.1 |
Yes | Yes |
| Scikit-learn | 1.0 | Yes | Yes |
| XGBoost | 1.5 | Yes | Yes |
| TensorFlow | 2.7 Training requires Watson Machine Learning Accelerator |
Yes | Yes |
| PyTorch | 1.10 Training requires Watson Machine Learning Accelerator |
Yes | Yes |
| Decision Optimization | 20.1 | No | Yes |
| Python function | 0.1 | yes | Programmatic only Inline payload only |
| Python scripts | 1.0 | No | Yes |
| RScript | 1.0 | No | Yes |
Note:
- If a framework is marked as deprecated then support for the framework will be removed in a future release.
Managing assets that refer to discontinued software or frameworks
When you have assets that rely on discontinued software specifications or frameworks, in some cases the migration is seamless. In other cases, your action is required to retrain or redeploy assets.
- Existing deployments of models built with discontinued framework versions or software specifications will be removed on the date of discontinuation.
- No new deployments of models built with discontinued framework versions or software specifications will be allowed.
For details on managing assets that rely on deprecated or discontinued frameworks and software specifications, refer to Managing outdated software specifications or frameworks.
Parent topic: Managing frameworks and software specifications