owen/lms
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added distance to previous in the recommendation outputs

Browse Source
This commit is contained in:
emeric
2026-06-10 18:38:05 +02:00
parent 1d75add3d8
commit a826a70c68
5 changed files with 54 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/libs/services/recommendation/impl/clusters/ClustersEngine.cpp
+5 -5
View File
@@ -86,7 +86,7 @@ namespace lms::recommendation
ResultContainer<IdType> res;
res.reserve(resultCount);
for (std::size_t i{}; i < resultCount; ++i)
res.push_back({ .id = overlapCounts[i].first, .distance = {} });
res.push_back({ .id = overlapCounts[i].first, .distanceToFirst = {}, .distanceToPrevious = {} });
return res;
}
@@ -295,7 +295,7 @@ namespace lms::recommendation
if (!bestIdx)
break;
res.push_back({ .id = candidates[*bestIdx], .distance = {} });
res.push_back({ .id = candidates[*bestIdx], .distanceToFirst = {}, .distanceToPrevious = {} });
selectedTracks.push_back(candidates[*bestIdx]);
candidates.erase(std::begin(candidates) + static_cast<std::ptrdiff_t>(*bestIdx));
}
@@ -364,7 +364,7 @@ namespace lms::recommendation
return {};
if (startTrackId == endTrackId)
return { RecommendationResult<db::TrackId>{ .id = startTrackId, .distance = {} } };
return { RecommendationResult<db::TrackId>{ .id = startTrackId, .distanceToFirst = {}, .distanceToPrevious = {} } };
db::Session& dbSession{ _db.getTLSSession() };
auto transaction{ dbSession.createReadTransaction() };
@@ -376,9 +376,9 @@ namespace lms::recommendation
TrackResults res;
res.reserve(std::min<std::size_t>(maxCount, 2));
res.push_back({ .id = startTrackId, .distance = {} });
res.push_back({ .id = startTrackId, .distanceToFirst = {}, .distanceToPrevious = {} });
if (maxCount > 1)
res.push_back({ .id = endTrackId, .distance = {} });
res.push_back({ .id = endTrackId, .distanceToFirst = {}, .distanceToPrevious = {} });
return res;
}