transformers-supporter 0.0.23

Transformers supporter

transformers-supporter

https://pypi.org/project/transformers-supporter

pip install transformers-supporter

Supported models

import transformers_supporter
transformers_supporter.register_auto()
transformers_supporter.register_pipeline()

Ann

tabular-regression

transformers_supporter.models.ann.configuration_ann.AnnConfig (transformers.AutoConfig)
transformers_supporter.models.ann.modeling_ann.AnnForTabularRegression
transformers_supporter.models.ann.feature_extraction_ann.TabularFeatureExtractor (transformers.AutoFeatureExtractor)
transformers_supporter.pipelines.tabular_regression_pipeline.TabularRegressionPipeline (pipeline(task="tabular-regression"))

tabular-classification

transformers_supporter.models.ann.configuration_ann.AnnConfig (transformers.AutoConfig)
transformers_supporter.models.ann.modeling_ann.AnnForTabularClassification
transformers_supporter.models.ann.feature_extraction_ann.TabularFeatureExtractor (transformers.AutoFeatureExtractor)
transformers_supporter.pipelines.tabular_regression_pipeline.TabularClassificationPipeline (pipeline(task="tabular-classification"))

transformers_supporter.models.AnnForTabularBinaryClassification

Cnn

image-classification

AutoConfig (transformers_supporter.models.CnnConfig)
AutoModelForImageClassification (transformers_supporter.models.CnnForImageClassification)
AutoImageProcessor (AutoImageProcessor)
pipeline(task="image-classification")

AutoConfig (transformers_supporter.models.CnnConfig)
AutoModelForImageClassification (transformers_supporter.models.CnnForImageClassification)
AutoImageProcessor (transformers_supporter.models.GrayscaleImageProcessor)
pipeline(task="image-classification")

object-detection

AutoConfig (transformers_supporter.models.FasterRcnnConfig)
AutoModelForObjectDetection (transformers_supporter.models.FasterRcnnForObjectDetection)
AutoImageProcessor (transformers_supporter.models.FasterRcnnImageProcessor)
pipeline(task="object-detection")

KeyPointDetection

transformers_supporter.models.CnnForKeyPointDetection

text-classification

AutoConfig (transformers_supporter.models.Embedded1dcnnConfig)
AutoModelForForSequenceClassification (transformers_supporter.models.Embedded1dcnnForSequenceClassification)
AutoFeatureExtractor (transformers_helper.models.TorchtextFeatureExtractor)
pipeline("text-classification")

Rnn

text-classification

AutoConfig (EmbeddedRnnConfig)
AutoModelForSequenceClassification (EmbeddedRnnForSequenceClassification)
AutoFeatureExtractor (TorchtextFeatureExtractor)

AutoConfig (PretrainedEmbeddedRnnConfig)
AutoModelForSequenceClassification (PretrainedEmbeddedRnnForSequenceClassification)
AutoFeatureExtractor (TorchtextFeatureExtractor)

audio-classification

AutoConfig (transformers_supporter.models.RnnConfig)
AutoModelForAudioClassification (transformers_supporter.models.RnnForAudioClassification)
AutoFeatureExtractor (Wav2Vec2, transformers_supporter.models.CustomWav2Vec2FeatureExtractor
pipeline(task="audio-classification")

TimeSeriesRegression

transformers_supporter.models.RnnForTimeSeriesRegression

translation

AutoConfig (transformers_helper.models.EmbeddedRNNConfig)
(transformers_supporter.models.EmbeddedRnnForFixedLengthTranslation)
AutoFeatureExtractor (transformers_helper.models.TorchtextTokenizer)
(transformers_supporter.pipelines.FixedLengthTranslationPipeline)

Custom Bert

text-classification

AutoConfig (transformers_supporter.models.CustomBertConfig)
AutoModelForSequenceClassification (transformers_supporter.models.CustomBertForSequenceClassification)
AutoTokenizer
pipeline(task="text-classification")

CustomImageClassificationPipeline

transformers_supporter.pipelines.CustomImageClassificationPipeline

TabularBinaryClassificationPipeline

transformers_supporter.pipelines.TabularBinaryClassificationPipeline