问题描述
似乎在 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 是专门为仅运行矩阵乘法等神经网络计算而设计的。