CODEWITHSUNDEEP
ruby-on-railsrubypostgresqlactiverecord
Alex Antonov
Alex Antonov

Reputation: 15176

Rails query random records using weights

let say I have a model Product

There are products

id | title   | weight (integer)
1  | Sugar   | 1
2  | Salt    | 1
3  | Pepper  | 2
4  | Coke    | 5

So, I want to get few (lets say 2) random products using weights.

Product.limit(2).rand_with_weights # =>
# The probability of coke is in array is 5x times bigger than salt or sugar and 2.5x times bigger than pepper
# The probability of pepper is in array is 2x times bigger than salt or sugar
# The probability of salt or sugar in array is equal

How can I make this query?

Upvotes: 4

Views: 627

Answers (1)

MHL
MHL

Reputation: 99

According to Efraimidis & Spirakis's paper: https://utopia.duth.gr/~pefraimi/research/data/2007EncOfAlg.pdf

If you are using postgres:

Product.order("RANDOM() ^ (1.0 / weight) DESC").limit(2)

proof of concept: http://sqlfiddle.com/#!17/474d2/4/0

As you can see the distribution is as expected

Upvotes: 2

Related Questions