Back to Home

NLP Pipeline

Build text classification, sentiment analysis, and entity extraction pipelines.

Text Classification

from oneml.nlp import TextClassifier

classifier = TextClassifier.from_preset("bert-base")
classifier.fit(train_texts, train_labels)
predictions = classifier.predict(["Great product!"])