Update README.md

README.md

@@ -21,8 +21,8 @@ mgen test --model SequenceClassification --model.backbone aido_rna_300m_mars --d
 ```python
 from modelgenerator.tasks import Embed
 model = Embed.from_config({"model.backbone": "aido_rna_300m_mars"}).eval()
-collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
-embedding = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["ACGT", "AGCT"]})
+embedding = model(transformed_batch)
 print(embedding.shape)
 print(embedding)
 ```
@@ -31,8 +31,8 @@ print(embedding)
 import torch
 from modelgenerator.tasks import SequenceClassification
 model = SequenceClassification.from_config({"model.backbone": "aido_rna_300m_mars", "model.n_classes": 2}).eval()
-collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
-logits = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["ACGT", "AGCT"]})
+logits = model(transformed_batch)
 print(logits)
 print(torch.argmax(logits, dim=-1))
 ```
@@ -41,8 +41,8 @@ print(torch.argmax(logits, dim=-1))
 import torch
 from modelgenerator.tasks import TokenClassification
 model = TokenClassification.from_config({"model.backbone": "aido_rna_300m_mars", "model.n_classes": 3}).eval()
-collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
-logits = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["ACGT", "AGCT"]})
+logits = model(transformed_batch)
 print(logits)
 print(torch.argmax(logits, dim=-1))
 ```
@@ -50,8 +50,8 @@ print(torch.argmax(logits, dim=-1))
 ```python
 from modelgenerator.tasks import SequenceRegression
 model = SequenceRegression.from_config({"model.backbone": "aido_rna_300m_mars"}).eval()
-collated_batch = model.collate({"sequences": ["ACGT", "AGCT"]})
-logits = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["ACGT", "AGCT"]})
+logits = model(transformed_batch)
 print(logits)
 ```
 
@@ -59,8 +59,8 @@ print(logits)
 ```python
 from genbio_finetune.tasks import Embed
 model = Embed.from_config({"model.backbone": "aido_rna_300m_mars"}).eval()
-collated_batch = model.collate({"sequences": ["ACGT", "ACGT"]})
-embedding = model(collated_batch)
+transformed_batch = model.transform({"sequences": ["ACGT", "ACGT"]})
+embedding = model(transformed_batch)
 print(embedding.shape)
 print(embedding)
 ```