Open fabiocat93 opened 6 months ago
Hi @fabiocat93 - I noticed the task is unassigned and would like to take a shot at it. Let me know if that works!
Hi @adi611 - Thanks for offering to take this on! I've assigned the task to you. Below are the details and the checklist to ensure everything meets the project requirements.
Your task involves implementing two main functionalities:
Sentiment Analysis: Determine the sentiment polarity of each text in the list, providing both a numerical score (ranging from -1 to 1) and a corresponding label ("negative", "neutral", or "positive").
Emotional Analysis: Identify the emotional content of each text, returning a dictionary of scores (probability for each possible emotion) and the dominant emotion as a string.
[ ] Implement Sentiment Analysis:
pieces_of_text: List[str]
, device: DeviceType
, model: SenselabModel
, and **args
as inputs.Dict[str, Any]
with the keys:
score
: A float between -1 and 1 indicating the sentiment polarity.label
: A string ("negative", "neutral", "positive") representing the sentiment category.[ ] Implement Emotional Analysis:
pieces_of_text: List[str]
, device: DeviceType
, model: SenselabModel
, and **args
as inputs.Dict[str, Any]
with the keys:
scores
: A dictionary with emotions as keys and their respective probabilities as values.dominant_emotion
: A string representing the most likely emotion in the text.score
: Float between -1 and 1.label
: String ("negative", "neutral", "positive").scores
: Dictionary with probabilities for each emotion.dominant_emotion
: String of the most probable emotion.I am happy to discuss this further if you feel it's needed. Thanks again for taking this on, and looking forward to your implementation!
Hey @fabiocat93, I have made some changes and added tests, but the commit needs me to run pytest on all the test cases, and it's taking too long at the test_synthesize_texts_with_mars5_model
. I checked the hardware requirements for the model and it’s way too much for my machine to handle. Any workaround?
Hey @fabiocat93, I have made some changes and added tests, but the commit needs me to run pytest on all the test cases, and it's taking too long at the
test_synthesize_texts_with_mars5_model
. I checked the hardware requirements for the model and it’s way too much for my machine to handle. Any workaround?
using SKIP=pytest
as a solution for now.
Description
Extracting sentiment (polarity) and emotional category from text
Tasks
Freeform Notes
No response