TensorFlow 1.4: как использовать BoostedTreesClassifier с Colab TPU

Question

У меня есть код для BoostedTreesClassifier, который работает, но занимает очень много времени с количеством данных, которые я передаю, и параметрами, которые я выбираю, т.е. max_depth https://www.tensorflow.org/api_docs/python/tf/estimator/BoostedTreesClassifier

Я пытаюсь использовать оценщик BoostedTreesClassifier в Colab с TPU, используя TPUEstimator https://www.tensorflow.org/api_docs/python/tf/contrib/tpu/TPUEstimator

Возможен ли BoostedTreesClassifier с TPUEstimator? Я вижу, что с оценщиком / TPUEstimator можно использовать только нейронные сети https://www.tensorflow.org/guide/using_tpu

Как правильно использовать BoostedTreesClassifier для работы с TPU Colab?

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

Answer 1

Я думаю, что использование стратегии TPUS является правильным подходом, но по некоторым причинам оно все еще занимает много времени.

import pandas as pd
import numpy as np
import tensorflow as tf
print(tf.__version__)


resolver = tf.distribute.cluster_resolver.TPUClusterResolver()
tf.tpu.experimental.initialize_tpu_system(resolver)
tpu_strategy = tf.distribute.experimental.TPUStrategy(resolver)

with tpu_strategy.scope():
  model = tf.estimator.BoostedTreesClassifier(
      feature_columns=attibute_columns,
      n_batches_per_layer=10,
      center_bias=True,
      n_trees=100,
      max_depth=20,
      pruning_mode='post',
      tree_complexity=0.1)

model.train(input_fn=train_input_fn)
results = model.evaluate(eval_input_fn)
print(results)