如何使用TPU运行tensorflow_dedecision_fores RandomForestModel



在TPU上运行时,RandomForestModel类似乎无法用于拟合,因为没有实现操作"SimpleMLNumericalFeature":

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

我发现错误:

NotFoundError:操作类型未在中注册"SimpleMLNumericalFeature"二进制在n-3f90423c-w-0上运行。确保Op和Kernel已在此进程中运行的二进制文件中注册。请注意,如果您从tf.contrib加载使用操作的已保存图形,访问(例如(tf.contrib.resampler应该在导入图形之前完成,因为首次访问模块时,contrib操作会延迟注册。[操作:__推理_训练_功能_833]

所以我想在没有TPU的情况下安装它。。。有可能做到吗?

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

谢谢!

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

相关内容

  • 没有找到相关文章

最新更新