balkian/senpy
1
0
Fork 0
mirror of https://github.com/gsi-upm/senpy synced 2024-11-22 00:02:28 +00:00
Code Issues Releases Wiki Activity

Fixed some code problems

Browse Source
This commit is contained in:
Daniel Suarez Souto 2018-09-28 11:36:53 +02:00
parent 38b478890b
commit 530982be62
1 changed files with 32 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

85
senpy/plugins/emotion/maxSentiment_plugin.py
View File

@ -17,36 +17,35 @@ class maxSentiment(AnalysisPlugin):
} }
def analyse_entry(self, entry, params): def analyse_entry(self, entry, params):
if params["max"]==True: if not params["max"]:
set_emotions= entry.emotions[0]['onyx:hasEmotion'] yield entry
max_emotion =set_emotions[0] return
# Extract max emotion from the set emotions (emotion with highest intensity)
for tmp_emotion in set_emotions:
if tmp_emotion['onyx:hasEmotionIntensity']>max_emotion['onyx:hasEmotionIntensity']:
max_emotion=tmp_emotion
if max_emotion['onyx:hasEmotionIntensity'] == 0: set_emotions= entry.emotions[0]['onyx:hasEmotion']
max_emotion['onyx:hasEmotionCategory'] = "neutral" max_emotion =set_emotions[0]
max_emotion['onyx:hasEmotionIntensity'] = 1.0
# Extract max emotion from the set emotions (emotion with highest intensity)
for tmp_emotion in set_emotions:
if tmp_emotion['onyx:hasEmotionIntensity']>max_emotion['onyx:hasEmotionIntensity']:
max_emotion=tmp_emotion
entry.emotions[0]['onyx:hasEmotion'] = [max_emotion] if max_emotion['onyx:hasEmotionIntensity'] == 0:
max_emotion['onyx:hasEmotionCategory'] = "neutral"
max_emotion['onyx:hasEmotionIntensity'] = 1.0
entry.emotions[0]['onyx:hasEmotion'] = [max_emotion]
entry.emotions[0]['prov:wasGeneratedBy'] = "maxSentiment" entry.emotions[0]['prov:wasGeneratedBy'] = "maxSentiment"
#print(entry) #print(entry)
yield entry yield entry
# Test Cases: # Test Cases:
# 1º Normal Situation. # 1 Normal Situation.
# 2º Case to return a Neutral Emotion. # 2 Case to return a Neutral Emotion.
test_cases = [{ test_cases = [{
"entry": { "entry": {
"@id": "#",
"@type": "entry", "@type": "entry",
"emotions": [ "emotions": [
{ {
@ -83,23 +82,16 @@ class maxSentiment(AnalysisPlugin):
"onyx:hasEmotionCategory": "disgust", "onyx:hasEmotionCategory": "disgust",
"onyx:hasEmotionIntensity": 0 "onyx:hasEmotionIntensity": 0
} }
], ]
'prov:wasGeneratedBy':"maxSentiment_plugin"
} },
], ],
"entities": [], "nif:isString": "This text makes me sad.\nwhilst this text makes me happy and surprised at the same time.\nI cannot believe it!"
"nif:isString": "This text makes me sad.\nwhilst this text makes me happy and surprised at the same time.\nI cannot believe it!",
"sentiments": [],
"suggestions": [],
"topics": []
}, },
'params': { 'params': {
'max': True 'max': True
}, },
'expected' : { 'expected' : {
"@id": "#",
"@type": "entry", "@type": "entry",
"emotions": [ "emotions": [
{ {
@ -112,19 +104,15 @@ class maxSentiment(AnalysisPlugin):
"onyx:hasEmotionCategory": "joy", "onyx:hasEmotionCategory": "joy",
"onyx:hasEmotionIntensity": 0.3333333333333333 "onyx:hasEmotionIntensity": 0.3333333333333333
} }
] ],
"prov:wasGeneratedBy" : 'maxSentiment'
} }
], ],
"entities": [], "nif:isString": "This text makes me sad.\nwhilst this text makes me happy and surprised at the same time.\nI cannot believe it!"
"nif:isString": "This text makes me sad.\nwhilst this text makes me happy and surprised at the same time.\nI cannot believe it!",
"sentiments": [],
"suggestions": [],
"topics": []
} }
}, },
{ {
"entry": { "entry": {
"@id": "#",
"@type": "entry", "@type": "entry",
"emotions": [ "emotions": [
{ {
@ -161,23 +149,17 @@ class maxSentiment(AnalysisPlugin):
"onyx:hasEmotionCategory": "disgust", "onyx:hasEmotionCategory": "disgust",
"onyx:hasEmotionIntensity": 0 "onyx:hasEmotionIntensity": 0
} }
], ]
'prov:wasGeneratedBy':"maxSentiment_plugin"
} }
], ],
"entities": [], "nif:isString": "This text makes me sad.\nwhilst this text makes me happy and surprised at the same time.\nI cannot believe it!"
"nif:isString": "This text makes me sad.\nwhilst this text makes me happy and surprised at the same time.\nI cannot believe it!",
"sentiments": [],
"suggestions": [],
"topics": []
}, },
'params': { 'params': {
'max': True 'max': True
}, },
'expected' : { 'expected' : {
"@id": "#",
"@type": "entry", "@type": "entry",
"emotions": [ "emotions": [
{ {
@ -190,14 +172,11 @@ class maxSentiment(AnalysisPlugin):
"onyx:hasEmotionCategory": "neutral", "onyx:hasEmotionCategory": "neutral",
"onyx:hasEmotionIntensity": 1 "onyx:hasEmotionIntensity": 1
} }
] ],
"prov:wasGeneratedBy" : 'maxSentiment'
} }
], ],
"entities": [], "nif:isString": "This text makes me sad.\nwhilst this text makes me happy and surprised at the same time.\nI cannot believe it!"
"nif:isString": "This text makes me sad.\nwhilst this text makes me happy and surprised at the same time.\nI cannot believe it!",
"sentiments": [],
"suggestions": [],
"topics": []
} }
}] }]