tensorflow - 如何使用 LabelBinarizer 解决 2 标签问题?

标签 tensorflow keras scikit-learn deep-learning

如果数据中只有 2 个标签,LabelBinarizer.fit_transform() 返回只有一列的数组。

但是对于我的用例的 tensorflow 模型训练,我需要的是标签数组中的 2 列。 如何使用 LabelBinarizer 来完成此操作,或者是否有其他 API?或者我是否需要通过迭代来手动修改数组?

最佳答案

由于您使用的是 TensorFlow,因此可以使用 tf.keras.utils.to_categorical .

关于tensorflow - 如何使用 LabelBinarizer 解决 2 标签问题?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58903111/

相关文章:

tensorflow - 机器学习: why the cost function does not need to be derivable?

python - tensorflow 中的自定义卷积函数

python - Scikit-learn 管道中的 Keras 模型与早期停止

python - 如何通过主题建模制作主题的百分比条形图?

tensorflow - 如何在TensorFlow中的TensorProto中访问tensor_content值?

tensorflow - 神经网络框架和 RL 算法库有什么区别?

python - 为什么这个 python 生成器根据 keras 没有输出?

python - TF.Keras 中自定义 Scratch 训练中的多输出多类分类

python - 通过 Sklearn 添加指标以在 Tensorflow 中进行预测测试

python - 使用 GridSearchCV 训练数据给我 ValueError,Sci-kit learn