mattwoodx commited on
Commit
4dd9188
1 Parent(s): bd6228a

Update readme

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -10,7 +10,7 @@ tags:
10
  library_name: transformers
11
  ---
12
  # Helix-mRNA-v0
13
-
14
  Helix-mRNA emerges as a hybrid state-space and transformer based model, leveraging both the efficient sequence processing capabilities of Mamba2's state-space architecture and the contextual understanding of transformer attention mechanisms, allowing for the best of both worlds between these two approaches. These traits make it particularly suitable for studying full-length transcripts, splice variants, and complex mRNA structural elements.
15
 
16
  We tokenize mRNA sequences at single-nucleotide resolution by mapping each nucleotide (A, C, U, G) and ambiguous base (N) to a unique integer. A further special character E is incorporated into the sequence, denoting the start of each codon. This fine-grained approach maximizes the model's ability to extract patterns from the sequences. Unlike coarser tokenization methods that might group nucleotides together or use k-mer based approaches, our single-nucleotide resolution preserves the full sequential information of the mRNA molecule. This simple yet effective encoding scheme ensures that no information is lost during the preprocessing stage, allowing the downstream model to learn directly from the raw sequence composition.
@@ -57,9 +57,12 @@ labels = [0, 2, 2, 0, 1]
57
  helixr_config = HelixmRNAConfig(batch_size=5, device=device, max_length=100)
58
  helixr_fine_tune = HelixmRNAFineTuningModel(helix_mrna_config=helixr_config, fine_tuning_head="classification", output_size=3)
59
 
 
60
  train_dataset = helixr_fine_tune.process_data(input_sequences)
61
 
 
62
  helixr_fine_tune.train(train_dataset=train_dataset, train_labels=labels)
63
 
 
64
  outputs = helixr_fine_tune.get_outputs(train_dataset)
65
  ```
 
10
  library_name: transformers
11
  ---
12
  # Helix-mRNA-v0
13
+
14
  Helix-mRNA emerges as a hybrid state-space and transformer based model, leveraging both the efficient sequence processing capabilities of Mamba2's state-space architecture and the contextual understanding of transformer attention mechanisms, allowing for the best of both worlds between these two approaches. These traits make it particularly suitable for studying full-length transcripts, splice variants, and complex mRNA structural elements.
15
 
16
  We tokenize mRNA sequences at single-nucleotide resolution by mapping each nucleotide (A, C, U, G) and ambiguous base (N) to a unique integer. A further special character E is incorporated into the sequence, denoting the start of each codon. This fine-grained approach maximizes the model's ability to extract patterns from the sequences. Unlike coarser tokenization methods that might group nucleotides together or use k-mer based approaches, our single-nucleotide resolution preserves the full sequential information of the mRNA molecule. This simple yet effective encoding scheme ensures that no information is lost during the preprocessing stage, allowing the downstream model to learn directly from the raw sequence composition.
 
57
  helixr_config = HelixmRNAConfig(batch_size=5, device=device, max_length=100)
58
  helixr_fine_tune = HelixmRNAFineTuningModel(helix_mrna_config=helixr_config, fine_tuning_head="classification", output_size=3)
59
 
60
+ # prepare data for input to the model
61
  train_dataset = helixr_fine_tune.process_data(input_sequences)
62
 
63
+ # fine-tune the model with the relevant training labels
64
  helixr_fine_tune.train(train_dataset=train_dataset, train_labels=labels)
65
 
66
+ # get outputs from the fine-tuned model on a processed dataset
67
  outputs = helixr_fine_tune.get_outputs(train_dataset)
68
  ```