Models

All registered Model components in DashAI.

Name	Description
ARIMA	Model a series as its own past values plus its own past errors.
AdaBoostClassifier	AdaBoost classifier that adapts to misclassified samples iteratively.
AdaBoostRegression	AdaBoost regressor that focuses on samples with high prediction errors.
AlbertTransformer	Pretrained ALBERT model for efficient English text classification.
BARTRegression	Bayesian Additive Regression Trees regressor.
BERTEmbedding	Dense embeddings using BERT models with configurable pooling.
BM25VectorizerModel	Model component that encapsulates a :class:CountVectorizer for BM25.
BagOfWordsTextClassificationModel	Text classification meta-model that combines a bag-of-words vectorizer with a DashAI tabular classifier.
BaggingClassifier	Bagging classifier that aggregates predictions from bootstrap subsets.
BayesianRidgeRegression	Bayesian Ridge regression with automatic regularisation estimation.
BertTransformer	Pretrained BERT model for English text classification.
BertinTransformer	Pretrained BERTIN model (Spanish RoBERTa) for Spanish text classification.
BetoTransformer	Pretrained BETO model for Spanish text classification.
CNNImageClassifier	CNN-based image classifier.
CustomAugmentationPrompt	User-defined augmentation prompt template for generating augmented retrieval prompts.
CustomRAGGenerationPrompt	CustomRAGGenerationPrompt class for user-defined prompt templates used in the language generation step of RAG.
DebertaV3Transformer	Pretrained DeBERTa-v3 transformer for text classification.
DecisionTreeClassifier	Decision tree classifier that learns axis-aligned decision rules from data.
DecisionTreeRegression	Decision tree regressor that recursively partitions the feature space.
DefaultAugmentationPrompt	Default prompt template for generating augmented retrieval queries.
DefaultQARAGGenerationPrompt	Default generation prompt for Question Answering tasks.
DefaultRAGGenerationPrompt	Default prompt template used in the language generation step of RAG.
DistilBERTEmbedding	Dense embeddings using DistilBERT models with configurable pooling.
DistilBertTransformer	Pretrained transformer DistilBERT allowing English text classification.
DummyClassifier	Baseline classifier that makes predictions ignoring the input features.
E5Embedding	Dense embeddings using E5 models with average pooling + L2 normalization.
EfficientNetB0ImageClassifier	EfficientNet-B0 image classifier (Tan & Le, 2019).
ElasticNetRegression	Elastic Net regression combining L1 and L2 penalties.
ElectraTransformer	Pretrained ELECTRA model for efficient English text classification.
ExponentialSmoothing	Forecast from a weighted average that favours recent observations.
ExtraTreesClassifier	Extra-Trees classifier using fully randomised decision tree splits.
ExtraTreesRegression	Extra-Trees regressor using fully randomised decision tree splits.
GaussianNB	Gaussian Naïve Bayes classifier based on Bayes' theorem.
GradientBoostingClassifier	Gradient boosting classifier that builds an ensemble of trees sequentially.
GradientBoostingR	Gradient boosting regressor that builds an ensemble of decision trees sequentially.
HistGradientBoostingClassifier	Histogram-based gradient boosting classifier for large datasets.
HistGradientBoostingRegression	Histogram-based gradient boosting regressor for large datasets.
InstructorEmbedding	Dense embeddings using INSTRUCTOR instruction-tuned models.
KNeighborsClassifier	K-nearest neighbours classifier that predicts the majority class among neighbours.
KNeighborsRegression	K-Nearest Neighbours regressor that averages the targets of nearest samples.
LaBSEmbedding	Dense embeddings using the LaBSE multilingual model (109 languages).
LassoRegression	Lasso regression with L1 regularisation for sparse coefficient solutions.
LeNet5ImageClassifier	LeNet-5 image classifier (LeCun et al.
LinearRegression	Ordinary least-squares linear regression model.
LinearSVCClassifier	Linear SVC classifier with Platt-scaling calibration for class probabilities.
LinearSVR	Support vector regression with a linear kernel for large datasets.
LogisticRegression	Logistic regression classifier with L1, L2, or Elastic-Net regularisation.
M2M100Transformer	M2M100 multilingual seq2seq model for configurable language-pair translation.
MLPClassifier	Multi-layer Perceptron classifier trained with backpropagation.
MLPImageClassifier	MLP-based image classifier.
MLPRegression	Single hidden-layer MLP regressor implemented in PyTorch.
MiniLMTransformer	Pretrained MiniLM model for lightweight English text classification.
ModernBertTransformer	Pretrained ModernBERT transformer for text classification.
MultilingualBertTransformer	Pretrained Multilingual BERT for crosslingual text classification.
NaiveForecaster	Predict that the series stays where it last was.
NllbTransformer	Pretrained transformer for configurable multilingual translation.
OpusMtEnDeTransformer	Pretrained transformer for English to German translation.
OpusMtEnESTransformer	Pretrained transformer for English to Spanish translation.
OpusMtEnFrTransformer	Pretrained transformer for English to French translation.
OpusMtEnRoaTransformer	Pretrained transformer for English to Romance translation.
OpusMtEsENTransformer	Pretrained transformer for Spanish to English translation.
OpusMtFrEnTransformer	Pretrained transformer for French to English translation.
OpusMtRoaEnTransformer	Pretrained transformer for Romance to English translation.
RandomForestClassifier	Random forest classifier that aggregates predictions from many decision trees.
RandomForestRegression	Random forest regressor that averages predictions from multiple decision trees.
ResNet18ImageClassifier	ResNet-18 image classifier (He et al.
ResNet50ImageClassifier	ResNet-50 image classifier (He et al.
RidgeRegression	Ridge regression with L2 regularisation to reduce coefficient magnitude.
RoBERTaEmbedding	Dense embeddings using RoBERTa / XLM-RoBERTa models with mean/max pooling.
RobertaTransformer	Pretrained RoBERTa model for English text classification.
SGDClassifier	SGD classifier with probability calibration for consistent predict_proba output.
SVC	Support vector machine classifier that maximises the margin between classes.
SVR	Support Vector Regressor using kernel-based function estimation.
SeasonalNaiveForecaster	Predict that each season repeats the one before it.
SentenceTransformerEmbedding	Dense embeddings using Sentence Transformer models.
T5SmallTransformer	T5-small seq2seq model for English-to-{German, French, Romanian} translation.
TFIDFVectorizerModel	Model component that encapsulates a :class:TfidfVectorizer.
TfIdfLogRegTextClassificationModel	TF-IDF vectorizer combined with Logistic Regression for text classification.
XlmRobertaTransformer	Pretrained XLM-RoBERTa model for multilingual text classification.
XlnetTransformer	Pretrained XLNet model for English text classification.