Add RandomesqueStrategy

Author: Frankie Robertson

Project.toml

@@ -23,6 +23,7 @@ MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 Mmap = "a63ad114-7e13-5084-954f-fe012c677804"
 PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
 PsychometricsBazaarBase = "b0d9cada-d963-45e9-a4c6-4746243987f1"
+QuickHeaps = "30b38841-0f52-47f8-a5f8-18d5d4064379"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
@@ -60,6 +61,7 @@ Mmap = "^1.11"
 Optim = "1.7.3"
 PrecompileTools = "1.2.1"
 PsychometricsBazaarBase = "^0.8.1"
+QuickHeaps = "0.2.2"
 Random = "^1.11"
 Reexport = "1"
 Setfield = "^1"

src/next_item_rules/NextItemRules.jl

@@ -65,6 +65,7 @@ include("./prelude/preallocate.jl")
 
 # Selection strategies
 include("./strategies/random.jl")
+include("./strategies/randomesque.jl")
 include("./strategies/sequential.jl")
 include("./strategies/exhaustive.jl")
 

src/next_item_rules/strategies/randomesque.jl

@@ -0,0 +1,53 @@
+using QuickHeaps: BinaryHeap, FastMax, Node, get_val
+using StatsBase: sample
+
+
+function randomesque(
+    rng::AbstractRNG,
+    objective::ItemCriterion,
+    responses::TrackedResponses,
+    items::AbstractItemBank,
+    k::Int
+)
+    objective_state = init_thread(objective, responses)
+    heap = BinaryHeap{Node{Int, Float64}}(o = FastMax)
+    sizehint!(heap, k)
+    for item_idx in eachindex(items)
+        if (findfirst(idx -> idx == item_idx, responses.responses.indices) !== nothing)
+            continue
+        end
+
+        obj_val = compute_criterion(objective, objective_state, responses, item_idx)
+
+        if length(heap) < k
+            push!(heap, Node(item_idx, obj_val))
+        elseif obj_val < get_val(peek(heap))
+            heap[1] = Node(item_idx, obj_val)
+        end
+    end
+    if length(heap) >= 1
+        Tuple(sample(rng, heap))
+    else
+        return (-1, Inf)
+    end
+end
+
+"""
+$(TYPEDEF)
+$(TYPEDFIELDS)
+
+"""
+struct RandomesqueStrategy <: NextItemStrategy
+    rng::AbstractRNG
+    k::Int
+end
+
+RandomesqueStrategy(k::Int) = RandomesqueStrategy(Xoshiro(), k)
+
+function best_item(
+    rule::ItemStrategyNextItemRule{RandomesqueStrategy, ItemCriterionT},
+    responses::TrackedResponses,
+    items
+) where {ItemCriterionT <: ItemCriterion}
+    randomesque(rule.rng, rule.criterion, responses, items, rule.k)[1]
+end