【问题标题】:How to use LabelBinarizer for 2 label problem?如何使用 LabelBinarizer 解决 2 个标签问题?
【发布时间】:2020-03-13 03:31:14
【问题描述】:

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

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

【问题讨论】:

    标签: tensorflow keras scikit-learn deep-learning


    【解决方案1】:

    由于你使用的是 TensorFlow,所以可以使用tf.keras.utils.to_categorical

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2013-04-30
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2016-04-28
      • 1970-01-01
      • 2020-03-27
      • 1970-01-01
      相关资源
      最近更新 更多