diff --git a/README.md b/README.md index b6f9919..9a1c85c 100644 --- a/README.md +++ b/README.md @@ -43,7 +43,7 @@ We can do this ```python from sequence_aligner.labelset import LabelSet from sequence_aligner.dataset import TrainingDataset -from sequence_aligner.containers import TraingingBatch +from sequence_aligner.containers import TrainingBatch import json raw = json.load(open('./data/ddi_train.json')) for example in raw: @@ -60,7 +60,7 @@ optimizer = AdamW(model.parameters(), lr=5e-6) dataloader = DataLoader( dataset, - collate_fn=TraingingBatch, + collate_fn=TrainingBatch, batch_size=4, shuffle=True, ) diff --git a/notebooks/Start Here.ipynb b/notebooks/Start Here.ipynb index fdd06b0..ee19bd5 100644 --- a/notebooks/Start Here.ipynb +++ b/notebooks/Start Here.ipynb @@ -27,7 +27,7 @@ "source": [ "from sequence_aligner.labelset import LabelSet\n", "from sequence_aligner.dataset import TrainingDataset\n", - "from sequence_aligner.containers import TraingingBatch\n", + "from sequence_aligner.containers import TrainingBatch\n", "import json\n" ] }, @@ -95,7 +95,7 @@ "\n", "dataloader = DataLoader(\n", " dataset,\n", - " collate_fn=TraingingBatch,\n", + " collate_fn=TrainingBatch,\n", " batch_size=4,\n", " shuffle=True,\n", ")\n", diff --git a/sequence_aligner/__init__.py b/sequence_aligner/__init__.py index e69de29..ae62ae6 100644 --- a/sequence_aligner/__init__.py +++ b/sequence_aligner/__init__.py @@ -0,0 +1,4 @@ +from .alignment import align_tokens_and_annotations_bilou +from .labelset import LabelSet +from .dataset import TrainingDataset +from .containers import TrainingBatch, TrainingExample diff --git a/sequence_aligner/containers.py b/sequence_aligner/containers.py index d2a32dd..4f96cc3 100644 --- a/sequence_aligner/containers.py +++ b/sequence_aligner/containers.py @@ -16,7 +16,7 @@ class TrainingExample: -class TraingingBatch: +class TrainingBatch: def __getitem__(self, item): return getattr(self, item)