w01fe · July 10, 2013 06:58
diff --git a/top_k.clj b/top_k.clj
 (defn top-k 
  "Return the top k elements of xs according to score-fn"
  [k score-fn xs]
  (let [n (count xs)
        scores (dbl/amake [i n] (score-fn (nth xs i)))
        ^ints idxs (dbl/amax-indices scores k)]
    (seq (hiphip.array/amake Object [i k] (nth xs (aget idxs (- n i 1)))))))
	(defn top-k
	"Return the top k elements of xs according to score-fn"
	[k score-fn xs]
	(let [n (count xs)
	scores (dbl/amake [i n] (score-fn (nth xs i)))
	^ints idxs (dbl/amax-indices scores k)]
	(seq (hiphip.array/amake Object [i k] (nth xs (aget idxs (- n i 1)))))))