标签: machine-learning catboost
我有一个用例,需要对非常大的数据集进行预测。因此,与训练相比,预测花费了将近10倍的时间。我想知道我是否可以使用GPU进行预测,或者它是否受CPU限制。我使用的是EC2 P3.16x大(8x Tesla V100,64个VCPU)
答案 0 :(得分:0)
我们还没有GPU预测器,只有测试集上的学习,度量计算和预测计算。
对于某些应用,Catboost可能会受益于GPU预测器(例如,许多决策树),但我们没有资源在不久的将来实现。