如何使用 TPU 运行 tensorflow_decision_forests RandomForestModel?

How to run a tensorflow_decision_forests RandomForestModel with TPU?

似乎 运行 在 TPU 上使用 RandomForestModel class 时无法用于拟合,因为未实现 op 'SimpleMLNumericalFeature' :

model_RF = tfdf.keras.RandomForestModel(task=tfdf.keras.Task.REGRESSION)
model_RF.fit(x=X_ent,y=Y_ent)

我收到错误:

NotFoundError: Op type not registered 'SimpleMLNumericalFeature' in binary running on n-3f90423c-w-0. Make sure the Op and Kernel are registered in the binary running in this process. Note that if you are loading a saved graph which used ops from tf.contrib, accessing (e.g.) tf.contrib.resampler should be done before importing the graph, as contrib ops are lazily registered when the module is first accessed. [Op:__inference_train_function_833]

所以我想在没有 TPU 的情况下安装它......可以这样做吗?

我需要启用 TPU,因为我想将此模型与 TPU 下 运行 的其他模型一起使用..

谢谢!

我认为在 TPU 上 运行 随机森林模型是不可能的,因为 TPU 专门设计用于 运行 矩阵乘法等神经网络计算。