R/which_max_random.R

In pomdp: Infrastructure for Partially Observable Markov Decision Processes (POMDP)

# this is borrowed from nnet::which.is.max
which.max.random <- function (x)
{
  mx <- x == max(x, na.rm = TRUE)
  mx[is.na(mx)] <- FALSE
  y <- seq_along(x)[mx]
  if (length(y) > 1L)
    sample(y, 1L)
  else
    y
}