Применить функцию confusionMatrix R к нескольким парам столбцов, сохраняя вывод в списках или tidy df - PullRequest
0 голосов
/ 25 сентября 2018

Я бы хотел применить функцию caretMatrixMatrix для нескольких столбцов, сохранив результаты в фрейме данных.

У меня есть два фрейма данных одного формата - "фактические данные" и "прогнозы".Они выглядят так:

x1   x2   x3   x4
 N    N    S    E
 E    W    E    E
 N    W    E    E

и т. Д. ... для многих строк, по 4 уровня в каждом столбце и довольно много столбцов

Я хотел бы сравнить соответствующие столбцыдва кадра данных с использованием confusionMatrix.Я могу сделать этот столбец за столбцом, например, начиная со столбца 1:

confusionMatrix(predictions[,1], actuals[,1])

, но вместо того, чтобы проходить каждый столбец один за другим, я хотел бы применить confusionMatrix ко всем соответствующим столбцам вдва фрейма данных.

В идеале я хотел бы, чтобы выходные данные были сохранены в каком-либо списке или фрейме данных, чтобы можно было легко извлекать определенные выходы из confusionMatrix (например, каппа или чувствительность) для всех сравнений.Я использовал purrr и broom, чтобы сделать что-то похожее с выходными данными линейных моделей, но входные данные здесь совершенно другие, и я не смог найти способ заставить это работать.

Большое спасибо

1 Ответ

0 голосов
/ 26 сентября 2018
a <- list()

for (i in 1:ncol(predictions))
    {
      a[[i]] <- confusionMatrix(predictions[,i], actuals[,i])
    }

print(a)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...