Emoji-AI-Avatar / avatar /__init__.py
Deminiko
Initial import: Emoji AI Avatar
25e624c
raw
history blame contribute delete
929 Bytes
"""
Avatar Module - Emotion detection and emoji mapping
Components:
- sentiment_transformer: Binary sentiment detection (DistilBERT) - positive/negative
- sentiment_multi_emotion: Multi-class emotion detection (RoBERTa) - 7 emotions
- sentiment_emoji_map: Emotion to emoji mapping
Usage:
# Binary model (faster, ~93% accuracy on polarity):
from avatar import SentimentAnalyzer
analyzer = SentimentAnalyzer()
# Multi-emotion model (more granular, 7 emotions):
from avatar import MultiEmotionAnalyzer
analyzer = MultiEmotionAnalyzer()
# Emoji mapping:
from avatar import EmojiMapper
mapper = EmojiMapper()
emoji = mapper.get_emoji("happiness")
"""
from .sentiment_transformer import SentimentAnalyzer
from .sentiment_multi_emotion import MultiEmotionAnalyzer
from .sentiment_emoji_map import EmojiMapper
__all__ = ["SentimentAnalyzer", "MultiEmotionAnalyzer", "EmojiMapper"]