mirror of
https://github.com/gsi-upm/senpy
synced 2024-11-14 12:42:27 +00:00
Added shelfmixin to emotion-wnaffect. This closes #2
This commit is contained in:
parent
bb0b0fadc2
commit
9ea177a780
@ -12,36 +12,12 @@ from nltk.corpus import WordNetCorpusReader
|
|||||||
|
|
||||||
from emotion import Emotion as Emo
|
from emotion import Emotion as Emo
|
||||||
from pattern.en import parse
|
from pattern.en import parse
|
||||||
from senpy.plugins import EmotionPlugin, SenpyPlugin
|
from senpy.plugins import EmotionPlugin, SenpyPlugin, ShelfMixin
|
||||||
from senpy.models import Results, EmotionSet, Entry, Emotion
|
from senpy.models import Results, EmotionSet, Entry, Emotion
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
class EmotionTextPlugin(EmotionPlugin):
|
class EmotionTextPlugin(EmotionPlugin, ShelfMixin):
|
||||||
|
|
||||||
def __init__(self, info, *args, **kwargs):
|
|
||||||
super(EmotionTextPlugin, self).__init__(info, *args, **kwargs)
|
|
||||||
self.id = info['module']
|
|
||||||
self.info = info
|
|
||||||
self._stopwords = stopwords.words('english')
|
|
||||||
local_path=os.path.dirname(os.path.abspath(__file__))
|
|
||||||
self._categories = {'anger': ['general-dislike',],
|
|
||||||
'fear': ['negative-fear',],
|
|
||||||
'disgust': ['shame',],
|
|
||||||
'joy': ['gratitude','affective','enthusiasm','love','joy','liking'],
|
|
||||||
'sadness': ['ingrattitude','daze','humility','compassion','despair','anxiety','sadness']}
|
|
||||||
|
|
||||||
self._wnaffect_mappings = {'anger': 'anger',
|
|
||||||
'fear': 'negative-fear',
|
|
||||||
'disgust': 'disgust',
|
|
||||||
'joy': 'joy',
|
|
||||||
'sadness': 'sadness'}
|
|
||||||
|
|
||||||
self._load_emotions(self.info['hierarchy_path'])
|
|
||||||
self._total_synsets = self._load_synsets(self.info['synsets_path'])
|
|
||||||
self._wn16_path = self.info['wn16_path']
|
|
||||||
self._wn16= None
|
|
||||||
self._wn16 = WordNetCorpusReader(os.path.abspath("{0}".format(self._wn16_path)), nltk.data.find(self._wn16_path))
|
|
||||||
|
|
||||||
|
|
||||||
def _load_synsets(self, synsets_path):
|
def _load_synsets(self, synsets_path):
|
||||||
@ -76,10 +52,41 @@ class EmotionTextPlugin(EmotionPlugin):
|
|||||||
Emo.emotions[name] = Emo(name, elem.get("isa"))
|
Emo.emotions[name] = Emo(name, elem.get("isa"))
|
||||||
|
|
||||||
def activate(self, *args, **kwargs):
|
def activate(self, *args, **kwargs):
|
||||||
|
self._stopwords = stopwords.words('english')
|
||||||
|
#local_path=os.path.dirname(os.path.abspath(__file__))
|
||||||
|
self._categories = {'anger': ['general-dislike',],
|
||||||
|
'fear': ['negative-fear',],
|
||||||
|
'disgust': ['shame',],
|
||||||
|
'joy': ['gratitude','affective','enthusiasm','love','joy','liking'],
|
||||||
|
'sadness': ['ingrattitude','daze','humility','compassion','despair','anxiety','sadness']}
|
||||||
|
|
||||||
|
self._wnaffect_mappings = {'anger': 'anger',
|
||||||
|
'fear': 'negative-fear',
|
||||||
|
'disgust': 'disgust',
|
||||||
|
'joy': 'joy',
|
||||||
|
'sadness': 'sadness'}
|
||||||
|
|
||||||
|
|
||||||
|
self._load_emotions(self._info['hierarchy_path'])
|
||||||
|
|
||||||
|
if 'total_synsets' not in self.sh:
|
||||||
|
total_synsets = self._load_synsets(self._info['synsets_path'])
|
||||||
|
self.sh['total_synsets'] = total_synsets
|
||||||
|
|
||||||
|
self._total_synsets = self.sh['total_synsets']
|
||||||
|
|
||||||
|
if 'wn16' not in self.sh:
|
||||||
|
self._wn16_path = self._info['wn16_path']
|
||||||
|
wn16 = WordNetCorpusReader(os.path.abspath("{0}".format(self._wn16_path)), nltk.data.find(self._wn16_path))
|
||||||
|
self.sh['wn16'] = wn16
|
||||||
|
|
||||||
|
self._wn16 = self.sh['wn16']
|
||||||
|
|
||||||
|
|
||||||
logger.info("EmoText plugin is ready to go!")
|
logger.info("EmoText plugin is ready to go!")
|
||||||
|
|
||||||
def deactivate(self, *args, **kwargs):
|
def deactivate(self, *args, **kwargs):
|
||||||
|
self.save()
|
||||||
logger.info("EmoText plugin is being deactivated...")
|
logger.info("EmoText plugin is being deactivated...")
|
||||||
|
|
||||||
def _my_preprocessor(self, text):
|
def _my_preprocessor(self, text):
|
||||||
|
@ -24,6 +24,7 @@
|
|||||||
"lxml>=3.4.2",
|
"lxml>=3.4.2",
|
||||||
"pandas",
|
"pandas",
|
||||||
"senpy",
|
"senpy",
|
||||||
"pattern"
|
"pattern",
|
||||||
|
"textblob"
|
||||||
]
|
]
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user