core.model.transformer

View Source
"""
# Transformer Models

Transformers based models, mostly based on
[BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding](https://arxiv.org/abs/1810.04805).

## Concrete Models
- `core.model.transformer.models.IsNextSCDBert`
- `core.model.transformer.models.IsSCDBert`
- `core.model.transformer.models.SelectSCDBert`
- `core.model.transformer.models.GivenTextFindSCDBert`, `core.model.transformer.models.GivenSCDFindTextBert`

### Dataset to use with Trainer
- `core.model.transformer.datasets.NextSentenceDataset`
- `core.model.transformer.datasets.ClassificationDataset`
- `core.model.transformer.datasets.MultipleChoiceDataset`
- `core.model.transformer.datasets.QuestionTextAnswerSCDDataset`, `core.model.transformer.datasets.QuestionSCDAnswerTextDataset`

"""

from core.model.transformer.datasets import (
	NextSentenceDataset,
	ClassificationDataset,
	MultipleChoiceDataset,
	QuestionTextAnswerSCDDataset, QuestionSCDAnswerTextDataset
)
from core.model.transformer.models import (
	IsNextSCDBert,
	IsSCDBert,
	SelectSCDBert,
	GivenTextFindSCDBert, GivenSCDFindTextBert
)