Функция вероятности Scala / Spark Softmax - PullRequest
Новая
       17

Функция вероятности Scala / Spark Softmax

0 голосов
/ 12 июля 2019

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

Например, 

val numbers = Seq(
       (.2, .7, .1),
       (.1, .9, 0),
       (1.0, 0, 0),
       (.4, .3, .3)
      ).toDF("A", "B", "C")

С тем, что я хочу вернуть как 

val numbers = Seq(
        (.2, .7, .1, "B"),
         (.1, .9, 0, "B"),
         (1.0, 0, 0, "A)",
         (.3, .3, .4, "C")
         ).toDF("A", "B", "C", "Highest_Class")
...