Skip to content

Commit

Permalink
fix: accidentally swapped training and query sets
Browse files Browse the repository at this point in the history
  • Loading branch information
nishaq503 committed Nov 12, 2023
1 parent d4808db commit 053a718
Showing 1 changed file with 15 additions and 15 deletions.
30 changes: 15 additions & 15 deletions cakes-results/src/genomic/read_silva.rs
Original file line number Diff line number Diff line change
Expand Up @@ -80,31 +80,31 @@ pub fn silva_to_dataset(
info!("Shuffled sequences and headers.");

// Split the lines into the training and query sets.
let queries = sequences.split_off(1000);
let (queries, query_headers): (Vec<_>, Vec<_>) = queries.into_iter().unzip();
let queries = VecDataset::new(format!("{stem}-queries"), queries, metric, is_expensive);
let query_headers = VecDataset::new(
format!("{stem}-query-headers"),
query_headers,
let train = sequences.split_off(1000);
let (train, train_headers): (Vec<_>, Vec<_>) = train.into_iter().unzip();
let train = VecDataset::new(format!("{stem}-queries"), train, metric, is_expensive);
let train_headers = VecDataset::new(
format!("{stem}-train-headers"),
train_headers,
metric,
is_expensive,
);
info!(
"Using {} sequences for queries.",
query_headers.cardinality()
"Using {} sequences for training.",
train_headers.cardinality()
);

let (train, train_headers): (Vec<_>, Vec<_>) = sequences.into_iter().unzip();
let train = VecDataset::new(format!("{stem}-train"), train, metric, is_expensive);
let train_headers = VecDataset::new(
format!("{stem}-train-headers"),
train_headers,
let (queries, query_headers): (Vec<_>, Vec<_>) = sequences.into_iter().unzip();
let queries = VecDataset::new(format!("{stem}-queries"), queries, metric, is_expensive);
let query_headers = VecDataset::new(
format!("{stem}-query-headers"),
query_headers,
metric,
is_expensive,
);
info!(
"Using {} sequences for training.",
train_headers.cardinality()
"Using {} sequences for queries.",
query_headers.cardinality()
);

assert_eq!(train.cardinality(), train_headers.cardinality());
Expand Down

0 comments on commit 053a718

Please sign in to comment.