Tensorflow.js中是否有类似于Python / Keras的to_categorical的实用程序函数?

问题描述

我最近开始使用tensorflow.js。我已经达到需要对输出进行热编码的地步了。我没有找到有关我的情况的tensorflow.js api( tensorflow api具有我作为实用程序正在搜索内容,但我正在使用javascript ),我从Google那里找到的唯一帮助是python / keras to_categorical()。

我想与tensorflow.js一起了解这个问题,因为我将基于python / keras to_categorical()源代码实现自己的函数,所以,我是否有类似的util函数可以直接从tensorflow.js api进行一种热编码吗?

解决方法

基于4.Pi.n注释-> one-hot,如果其他人不知道如何正确搜索文档。