上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章将研究其他选项。