neuralbioinfo
/

prokbert-mini-phage

Text Classification

sequence embedding

genomic language models

promoter-prediction

Model card Files Files and versions

ligeti commited on Feb 16, 2024

Commit

4732d24

·

verified ·

1 Parent(s): 2055e51

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -44,7 +44,7 @@ shift= 1
 tok_params = {'kmer' : kmer,
              'shift' : shift}
 tokenizer = ProkBERTTokenizer(tokenization_params=tok_params)
-model = BertForBinaryClassificationWithPooling.from_pretrained(finetuned_model)
 sequence = 'CACCGCATGGAGATCGGCACCTACTTCGACAAGCTGGAGGCGCTGCTGAAGGAGTGGTACGAGGCGCGCGGGGGTGAGGCATGACGGACTGGCAAGAGGAGCAGCGTCAGCGC'
 inputs = tokenizer(sequence, return_tensors="pt")
 # Ensure that inputs have a batch dimension

 tok_params = {'kmer' : kmer,
              'shift' : shift}
 tokenizer = ProkBERTTokenizer(tokenization_params=tok_params)
+model = MegatronBertForSequenceClassification.from_pretrained(finetuned_model)
 sequence = 'CACCGCATGGAGATCGGCACCTACTTCGACAAGCTGGAGGCGCTGCTGAAGGAGTGGTACGAGGCGCGCGGGGGTGAGGCATGACGGACTGGCAAGAGGAGCAGCGTCAGCGC'
 inputs = tokenizer(sequence, return_tensors="pt")
 # Ensure that inputs have a batch dimension