Skip to content

Commit

Permalink
Shuffle queries even more
Browse files Browse the repository at this point in the history
  • Loading branch information
stsievert committed Jul 1, 2021
1 parent 4865c36 commit 51a6725
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion salmon/triplets/samplers/_adaptive_runners.py
Original file line number Diff line number Diff line change
Expand Up @@ -348,7 +348,7 @@ class ARR(Adaptive):
References
----------
.. [1] Heim, Eric, et al. "Active perceptual similarity modeling withi
.. [1] Heim, Eric, et al. "Active perceptual similarity modeling with
auxiliary information." arXiv preprint arXiv:1511.02254 (2015). https://arxiv.org/abs/1511.02254
"""
Expand Down Expand Up @@ -382,6 +382,7 @@ def get_queries(self, *args, **kwargs):

top_queries = df.loc[top_idx]
top_scores = top_queries["score"].to_numpy()
top_queries = top_queries.sample(frac=1, replace=False)

posted = top_queries[["h", "l", "r"]].to_numpy().astype("int64")
r_scores = np.random.uniform(low=10, high=11, size=len(posted))
Expand Down

0 comments on commit 51a6725

Please sign in to comment.