如果数据中只有 2 个标签,LabelBinarizer.fit_transform() 返回只有一列的数组。
但是对于我的用例的 tensorflow 模型训练,我需要的是标签数组中的 2 列。 如何使用 LabelBinarizer 来完成此操作,或者是否有其他 API?或者我是否需要通过迭代来手动修改数组?
最佳答案
由于您使用的是 TensorFlow,因此可以使用 tf.keras.utils.to_categorical .
关于tensorflow - 如何使用 LabelBinarizer 解决 2 标签问题?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/58903111/