modbot.training.models

Models

Classes

BERT([texts, clf, vec])

Bert model

BertCNN([texts, clf, vec])

Moderation model with BERT and CNN

BertCnnLstm([texts, clf, vec])

Model with Bert, CNN, and LSTM in that order

BertCnnTorch(*args, **kwargs)

Bert Cnn model pytorch implementation

BertCnnTorchModel(embed_size)

Bert Cnn model pytorch implementation

BertCnnTorchModelSmall(embed_size)

Bert Cnn model pytorch implementation

BertCnnTorchSmall(*args, **kwargs)

Bert Cnn model pytorch implementation

BertLSTM([texts, clf, vec])

Bert model with LSTM

BertLstmCnn([texts, clf, vec])

Model with Bert, LSTM, and CNN in that order

BertTorch(*args, **kwargs)

Bert model pytorch implementation

BertTorchModel(embed_size)

Bert model pytorch implementation

CNN([texts, clf, vec])

Moderation model using convolution layer

KMeansComponent()

KMeans model class.

LSTM([texts, clf, vec])

LSTM model for moderation

ModerationModel([texts, model])

Base moderation model class

NNmodel([texts, clf, vec])

Base keras model for moderation

SVM([texts, model])

Linear SVM model

TorchModel([checkpoint, embed_size, lr, device])

Base torch model