TensorFlow 1.4：如何在Colab TPU中使用BoostedTreesClassifier

Question

我有BoostedTreesClassifier的代码，该代码可以正常工作，但是要花费很长时间，因为我要馈送的数据量以及我选择的参数即max_depth https://www.tensorflow.org/api_docs/python/tf/estimator/BoostedTreesClassifier

我正在尝试通过TPUEstimator在Colab中将BoostedTreesClassifier估计器与TPU一起使用 https://www.tensorflow.org/api_docs/python/tf/contrib/tpu/TPUEstimator

TPedstimator是否可以使用BoostedTreesClassifier？我看到只有神经网络才能与Estimator / 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)

Answer 1

我认为使用TPUStrategy是正确的方法，但是由于某些原因，它仍然需要花费很长时间。

object