osmarks/meme-search-engine
1
0
Fork 0
mirror of https://github.com/osmarks/meme-search-engine.git synced 2025-10-14 15:17:42 +00:00
Code Issues Releases Wiki Activity

integrate rating model correctly

Browse Source
This commit is contained in:
osmarks
2025-01-23 13:45:59 +00:00
parent fbf9b3421d
commit 3852d0078d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/dump_processor.rs
View File

@@ -451,7 +451,7 @@ fn main() -> Result<()> {
for (i, (x, _embedding)) in batch.into_iter().enumerate() { for (i, (x, _embedding)) in batch.into_iter().enumerate() {
let (vertices, shards) = read_out_vertices(count + i as u32)?; // TODO: could parallelize this given the batching let (vertices, shards) = read_out_vertices(count + i as u32)?; // TODO: could parallelize this given the batching
let mut entry_scores = scores[i..(i + score_model.output_channels)].to_vec(); let mut entry_scores = scores[(i * score_model.output_channels)..((i + 1) * score_model.output_channels)].to_vec();
entry_scores.push(x.timestamp as f32); // seconds since epoch, so precision issues aren't awful entry_scores.push(x.timestamp as f32); // seconds since epoch, so precision issues aren't awful