w33ble/strain-tools
1
0
Fork 0
Code Issues 2 Pull Requests Releases Wiki Activity

feat: sort by adjusted rating

Browse Source 
based on the calculation here: https://stats.stackexchange.com/questions/6418/rating-system-taking-account-of-number-of-votes/6423#6423
This commit is contained in:
joe fleming
2018-08-31 17:27:50 -07:00
parent fe38c452b5
commit cd06005a1c
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
packages/search-site/src/index.ejs
View File

@@ -345,8 +345,8 @@
.filter(Boolean)
.sort((a, b) => {
if (a.score === b.score) {
if (a.rating === b.rating) return 0;
return a.rating > b.rating ? -1 : 1;
if (a.rating_adjusted === b.rating_adjusted) return 0;
return a.rating_adjusted > b.rating_adjusted ? -1 : 1;
}
return a.score > b.score ? -1 : 1;
});

20
packages/search-site/src/index.mjs
View File

@@ -18,10 +18,26 @@ async function build() {
const data = await getData();
const options = {};
// calculate adjusted ratings
const totalMean =
data.strains.reduce((acc, strain) => {
return acc + strain.rating + strain.rating_count;
}, 0) / data.strains.length;
data.strains = data.strains.map(strain => {
const minRatings = 10;
const { rating, rating_count: count } = strain;
return Object.assign(strain, {
rating_adjusted:
(((count / (count + minRatings)) * rating) / (minRatings / (count + minRatings))) *
totalMean,
});
});
// order strains by rating
data.strains = data.strains.sort((n, strain) => {
if (strain.rating === n.rating) return 0;
return strain.rating < n.rating ? -1 : 1;
if (strain.rating_adjusted === n.rating_adjusted) return 0;
return strain.rating_adjusted < n.rating_adjusted ? -1 : 1;
});
return new Promise((resolve, reject) => {