From ee75eea939e3fd0b1e790655bdc0f9ac2a4b822d Mon Sep 17 00:00:00 2001 From: Andrew Lee Date: Sat, 25 Apr 2020 17:38:21 -0400 Subject: [PATCH] Changed interval check --- src/routes/postSkipSegments.js | 19 ++++++++----------- 1 file changed, 8 insertions(+), 11 deletions(-) diff --git a/src/routes/postSkipSegments.js b/src/routes/postSkipSegments.js index cab50c0..9c9b5a2 100644 --- a/src/routes/postSkipSegments.js +++ b/src/routes/postSkipSegments.js @@ -114,19 +114,16 @@ async function autoModerateSubmission(submission, callback) { let overlap = false; http = await fetch("https://ai.neuralblock.app/api/getSponsorSegments?vid=" + submission.videoID); if (http.status === 500) return false; - nb_predictions = await http.json(); + nb_predictions = await http.json(); for (const nb_seg of nb_predictions.sponsorSegments){ - let head = 0; - let tail = 0; - // If there's an overlap, find the percentage of overlap. - if (submission.startTime <= nb_seg[1] && nb_seg[0] <= submission.endTime){ - head = Math.max(submission.startTime, nb_seg[0]); - tail = Math.min(submission.endTime, nb_seg[1]); - } - if ((tail-head)/(nb_seg[1]-nb_seg[0]) >= 0.65){ - overlap = true; - break; + // The submission needs to be a subset of the widened NB prediction + // and at least 65% of NB's prediction. + if (nb_seg[0]-2 <= submission.startTime && submission.endTime <= nb_seg[1]+2){ + if ((submission.endTime - submission.startTime)/(nb_seg[1]-nb_seg[0]) >= 0.65){ + overlap=true; + break; + } } } if (overlap){