Kubeflow学习指南:生产级机器学习系统实现
上QQ阅读APP看书,第一时间看更新

1.5.3 训练

特征准备好后,你就可以建立和训练模型了。Kubeflow支持多种分布式训练框架。Kubeflow目前支持:

·TensorFlow(https://www.tensorflow.org

·PyTorch(https://pytorch.org

·Apache MXNet(https://mxnet.apache.org

·XGBoost(https://github.com/dmlc/xgboost

·Chainer(https://chainer.org

·Caffe2(https://caffe2.ai

·消息传递接口(Message Passing Interface,MPI)(https://oreil.ly/0zln4

在第7章中,我们将更详细地研究Kubeflow如何训练TensorFlow模型,第9章将研究其他选项。