Skip to content

Commit 05bc6a8

Browse files
authored
Update predict_parquet.py to ignore the sparse tiles in bd and tx
1 parent eb5f40c commit 05bc6a8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/segger/prediction/predict_parquet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -329,7 +329,7 @@ def _get_id():
329329
transcript_id = batch["tx"].id.cpu().numpy().astype("str")
330330
assignments = {"transcript_id": transcript_id}
331331

332-
if len(batch["bd"].pos) >= 10:
332+
if len(batch["bd"].pos) >= 10 and len(batch["tx"].pos) >= 1000:
333333
# Step 1: Compute similarity scores between 'tx' (transcripts) and 'bd' (boundaries)
334334
scores = get_similarity_scores(
335335
lit_segger.model, batch, "tx", "bd", receptive_field, knn_method=knn_method, gpu_id=gpu_id

0 commit comments

Comments
 (0)