Отображение соответствующего классификатора sklearn из числовых (категориальных) прогнозов в одномерные прогнозы

#python #machine-learning #scikit-learn #one-hot-encoding

Вопрос:

Установка:

В рамках эксперимента я использую две библиотеки, скажем A , и B . Библиотека A обучает двоичный классификатор (предпочтительно классификатор sklearn), а библиотека B берет предварительно обученный двоичный классификатор и изменяет его. Конвейер этого эксперимента состоит в том, чтобы обучить классификатор с использованием библиотеки A , а затем изменить этот же классификатор с помощью библиотеки B . Единственная проблема с этим процессом заключается в том, что библиотека A ожидала, что классификатор будет выводить прогнозы в формате 1D (т. Е. предсказывать либо 1 или 0 ), в то время как библиотека B ожидает, что классификатор будет выводить прогнозы в формате 2D или в одной горячей кодировке (т. е. предсказывать либо [0, 1] или [1, 0] ).

Вопрос:

Существует ли метод для взятия соответствующего классификатора sklearn, который делает 1D двоичные предсказания, и изменения его таким образом, чтобы его выходы были 2D двоичными предсказаниями (т. Е. предсказаниями с одним горячим кодированием) без переподготовки классификатора.