как переупорядочить таблицу сопряженности для формирования матрицы путаницы в R

Question

С точки зрения применения алгоритма кластеризации, после того, как я получил таблицу вероятностей для каждой комбинации фактического класса и прогнозируемого класса, например,

  1  2  3
A 2  3  15
B 20 1  4
C 0  32 1

Как мне написать функцию для получить матрицу путаницы, которая максимизирует диагональ путем изменения порядка столбцов? Спасибо!

  1  2  3
A 15 2  3
B 4  20 1
C 1  0  32

Answer 1

Мы могли бы использовать max.col, чтобы получить индекс столбца максимального значения для строки, и использовать его для перестановки столбцов

m2 <- m1[,max.col(m1, 'first')]
colnames(m2) <- seq_len(ncol(m2))
m2
#   1  2  3
#A 15  2  3
#B  4 20  1
#C  1  0 32

data

m1 <- structure(c(2L, 20L, 0L, 3L, 1L, 32L, 15L, 4L, 1L), .Dim = c(3L, 
  3L), .Dimnames = list(c("A", "B", "C"), c("1", "2", "3")))