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

Question

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

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

标签： tensorflowkerasscikit-learndeep-learning

Answer 1

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