【发布时间】:2020-03-13 03:31:14
【问题描述】:
如果数据中只有 2 个标签,LabelBinarizer.fit_transform() 返回只有一列的数组。
但是对于我的用例的 tensorflow 模型训练,我需要的是标签数组中的 2 列。 如何使用 LabelBinarizer 完成此操作,或者是否有任何其他 API?或者我是否需要通过迭代来手动修改数组?
【问题讨论】:
标签: tensorflow keras scikit-learn deep-learning
如果数据中只有 2 个标签,LabelBinarizer.fit_transform() 返回只有一列的数组。
但是对于我的用例的 tensorflow 模型训练,我需要的是标签数组中的 2 列。 如何使用 LabelBinarizer 完成此操作,或者是否有任何其他 API?或者我是否需要通过迭代来手动修改数组?
【问题讨论】:
标签: tensorflow keras scikit-learn deep-learning
由于你使用的是 TensorFlow,所以可以使用tf.keras.utils.to_categorical。
【讨论】: