xgboost是否会使用 "gpu-hist "的TPU（如果有TPU的话）？

Question

我很好奇，如果有TPU的话，xgboost是否会在google colab中使用TPU？当然，它可以很好地利用GPU...

Answer 1

谢谢你的问题。TPU通常在基于神经网络的模型上非常出色。我没有听说过在TPU上进行基于树的计算的方法，我非常怀疑TPU在今天的情况下会有这样的表现。