raghubhetwal / email-_categorization

AI Assisted Message Distribution :: Objective is to automatically predict the top 5 recipients for incoming emails based on the email body.

Email-_Categorization

AI Assisted Message Distribution :: Objective is to automatically predict the top 5 recipients for incoming emails based on the email body. Dataset used from Enron Email Data. BERT, NB and LSTM were used for creation of Model.

Flow of ML process

Training and Evaluation

Result

Prediction

The LSTM, BERT model, trained on pre-processed data and leveraging sequential modeling, achieved a high level of accuracy;

Future work could involve further optimization of the model, exploring alternative algorithms to extract more relevant information with sentiment analysis, and evaluating the system on different datasets!