From 3852d0078d8df4696d3e9327270c2f752bbb11c1 Mon Sep 17 00:00:00 2001 From: osmarks Date: Thu, 23 Jan 2025 13:45:59 +0000 Subject: [PATCH] integrate rating model correctly --- src/dump_processor.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/dump_processor.rs b/src/dump_processor.rs index ea34066..38d9b2d 100644 --- a/src/dump_processor.rs +++ b/src/dump_processor.rs @@ -451,7 +451,7 @@ fn main() -> Result<()> { 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 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