Сравните заголовок столбца в горячем коде и прогнозируемые метки - PullRequest
1 голос
/ 29 марта 2020

У меня есть 3 столбца с горячим кодированием, в которых названия заголовков являются метками, и один столбец прогнозирования preds, где метки прогнозируются (см. Изображение). Я хочу рассчитать эффективность моих прогнозов, сравнив метку в preds и заголовок столбца с 1 кодировкой.

see here

В этом примере у меня правильно прогнозируется только 20% .

Есть ли быстрый способ сделать это в pandas?

1 Ответ

1 голос
/ 29 марта 2020

IIU C, DataFrame.lookup и np.mean

df[['Type_1','Type_2','Type_3']].lookup(df.index, df['preds']).mean() * 100
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...