问Colab 1.4:如何将BoostedTreesClassifier与TensorFlow TPU一起使用
EN

Stack Overflow用户

提问于 2019-06-25 03:41:33

回答 1查看 213关注 0票数 0

我有BoostedTreesClassifier的代码，它可以工作，但需要很长时间的数据量，我给它和参数我选择，即max_depth https://www.tensorflow.org/api_docs/python/tf/estimator/BoostedTreesClassifier

我正在尝试使用TPUEstimator https://www.tensorflow.org/api_docs/python/tf/contrib/tpu/TPUEstimator在具有TPU的Colab中使用BoostedTreesClassifier estimator

TPUEstimator可以使用BoostedTreesClassifier吗？我发现只有神经网络可以与估计器/TPUEstimator https://www.tensorflow.org/guide/using_tpu一起使用

让BoostedTreesClassifier与Colab TPU协同工作的正确方法是什么？

tpu_estimator = tf.contrib.tpu.TPUEstimator(
    model_fn=model_fn,
    config=my_tpu_run_config,
    train_batch_size=100,
    use_tpu=True)

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/56742965

复制

相似问题

问Colab 1.4:如何将BoostedTreesClassifier与TensorFlow TPU一起使用EN