perf: Optimized analyze_text function for faster execution
Browse files
src/SentimentAndIntentionAnalysis.py
CHANGED
|
@@ -6,6 +6,7 @@ class ZeroShotClassifier:
|
|
| 6 |
self.model_name = model_name
|
| 7 |
self.sentiment_labels = ["Positive", "Negative", "Neutral"]
|
| 8 |
self.intention_labels = ["Inquire", "Inform", "Payment", "Price", "Trade In", "Discount", "Complaint", "Approve", "Selling", "Confusion", "Change Package", "Upgrade", "Purchase", "Help"]
|
|
|
|
| 9 |
|
| 10 |
def create_model(self, model_name):
|
| 11 |
# Create Model
|
|
|
|
| 6 |
self.model_name = model_name
|
| 7 |
self.sentiment_labels = ["Positive", "Negative", "Neutral"]
|
| 8 |
self.intention_labels = ["Inquire", "Inform", "Payment", "Price", "Trade In", "Discount", "Complaint", "Approve", "Selling", "Confusion", "Change Package", "Upgrade", "Purchase", "Help"]
|
| 9 |
+
self.labels = self.sentiment_labels + self.intention_labels
|
| 10 |
|
| 11 |
def create_model(self, model_name):
|
| 12 |
# Create Model
|