14 lines
580 B
Python
Executable file
14 lines
580 B
Python
Executable file
from obsei.analyzer.classification_analyzer import ClassificationAnalyzerConfig, ZeroShotClassificationAnalyzer
|
|
|
|
# initialize classification analyzer config
|
|
# It can also detect sentiments if "positive" and "negative" labels are added.
|
|
analyzer_config=ClassificationAnalyzerConfig(
|
|
labels=["service", "delay", "performance"],
|
|
)
|
|
|
|
# initialize classification analyzer
|
|
# For supported models refer https://huggingface.co/models?filter=zero-shot-classification
|
|
analyzer = ZeroShotClassificationAnalyzer(
|
|
model_name_or_path="typeform/mobilebert-uncased-mnli",
|
|
device="auto"
|
|
)
|