如何使用 TPU 运行 tensorflow_decision_forests RandomForestModel?

问题描述

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

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

我发现了错误

NotFoundError: Op 类型未注册 'SimpleMLNumericalFeature' 在 n-3f90423c-w-0 上运行的二进制文件。确保 Op 和 Kernel 是 在此进程中运行的二进制文件注册。请注意,如果您是 加载使用来自 tf.contrib 的操作的已保存图形,访问(例如) tf.contrib.resampler 应该在导入图形之前完成,因为 当第一次访问模块时,contrib 操作被延迟注册。 [操作:__inference_train_function_833]

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

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

谢谢!

解决方法

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