Spaces:
Sleeping
Sleeping
File size: 4,886 Bytes
267744b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 |
import torch
from transformers import BertTokenizer, BertForSequenceClassification, DistilBertForSequenceClassification
from datetime import datetime
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# Load the intent classifier model and tokenizer
num_intent_labels = 151 # Set the correct number of labels for the intent classifier
intent_model = BertForSequenceClassification.from_pretrained("bert-base-uncased", num_labels=num_intent_labels)
intent_model.load_state_dict(torch.load("intent_classifier.pth"))
intent_tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
intent_model.to(device)
intent_model.eval()
# Load the emotions model and tokenizer
emotions_model = DistilBertForSequenceClassification.from_pretrained("./saved_model")
emotions_tokenizer = BertTokenizer.from_pretrained("./saved_model")
emotions_model.to(device)
emotions_model.eval()
# Define the label names for emotions
emotion_label_names = ["admiration", "amusement", "anger", "annoyance", "approval", "caring", "confusion", "curiosity", "desire", "disappointment", "disapproval", "disgust", "embarrassment", "excitement", "fear", "gratitude", "grief", "joy", "love", "nervousness", "optimism", "pride", "realization", "relief", "remorse", "sadness", "surprise", "neutral"]
def predict_intent(sentence):
inputs = intent_tokenizer(sentence, return_tensors="pt", padding="max_length", truncation=True, max_length=128)
inputs = {key: val.to(device) for key, val in inputs.items()}
with torch.no_grad():
outputs = intent_model(**inputs)
predicted_class = torch.argmax(outputs.logits, dim=1).cpu().numpy()[0]
return predicted_class
def predict_emotion(sentence):
inputs = emotions_tokenizer(sentence, return_tensors="pt", padding="max_length", truncation=True, max_length=128)
inputs = {key: val.to(device) for key, val in inputs.items() if key != "token_type_ids"}
with torch.no_grad():
outputs = emotions_model(**inputs)
predicted_class = torch.argmax(outputs.logits, dim=1).cpu().numpy()[0]
return predicted_class, emotion_label_names[predicted_class]
def calculate_priority_score(intent, emotion, time_remaining):
# Example priority score calculation
intent_weight = 0.4
emotion_weight = 0.3
time_weight = 0.3
# Normalize time_remaining to a score between 0 and 1
time_score = max(0, min(1, 1 - (time_remaining.total_seconds() / (24 * 3600))))
# Calculate priority score
priority_score = (intent * intent_weight) + (emotion * emotion_weight) + (time_score * time_weight)
return priority_score
def prioritize_task(task_description, due_date_time, predicted_emotion, predicted_label_name):
predicted_intent = predict_intent(task_description)
# Calculate time remaining until the due date and time
due_date_time = datetime.strptime(due_date_time, "%Y-%m-%d %H:%M:%S")
time_remaining = due_date_time - datetime.now()
priority_score = calculate_priority_score(predicted_intent, predicted_emotion, time_remaining)
return {
"description": task_description,
"due_date_time": due_date_time,
"time_remaining": time_remaining,
"predicted_intent": predicted_intent,
"predicted_emotion": predicted_emotion,
"predicted_label_name": predicted_label_name,
"priority_score": priority_score
}
# Example tasks
tasks = [
{"description": "Finish the report by tomorrow.", "due_date_time": "2025-03-02 09:00:00"},
{"description": "meeting", "due_date_time": "2025-03-02 12:00:00"},
{"description": "listen to music.", "due_date_time": "2025-03-02 15:00:00"},
{"description": "daily linkedin queens game.", "due_date_time": "2025-03-02 18:00:00"},
{"description": "prepare ppt", "due_date_time": "2025-03-02 21:00:00"}
]
# Overall emotion sentence
emotion_sentence = "I am feeling very tired and stressed now"
predicted_emotion, predicted_label_name = predict_emotion(emotion_sentence)
# Prioritize tasks
prioritized_tasks = []
for task in tasks:
prioritized_tasks.append(prioritize_task(task["description"], task["due_date_time"], predicted_emotion, predicted_label_name))
# Reorder tasks based on priority score (descending order)
prioritized_tasks.sort(key=lambda x: x["priority_score"], reverse=True)
# Print prioritized tasks
for task in prioritized_tasks:
print(f"Task Description: '{task['description']}'")
print(f"Due Date and Time: {task['due_date_time']}")
print(f"Time Remaining: {task['time_remaining']}")
print(f"Predicted Intent: {task['predicted_intent']}")
print(f"Predicted Emotion: {task['predicted_emotion']} ({task['predicted_label_name']})")
print(f"Priority Score: {task['priority_score']:.4f}")
print()
|