AJAYNAIDU33/Email-Classification
0
1�
2 � h* � �` � d dl Z d dlmZ e j d� � Z e j d� � Zd� ZdS )� N)�mask_piizemail_classifier_model.pklztfidf_vectorizer.pklc � � t | � � \ }}t � |g� � }t � |� � d }| |||d�S )Nr )�input_email_body�list_of_masked_entities� masked_email�category_of_the_email)r �3vectorizer� transform�model�predict)�4email_bodyr � entity_list� input_vector�predicted_categorys �@C:\Users\ajayn\OneDrive\Documents\email-classification\models.py�classify_emailr s] � � (�� 4� 4� �L�+� �'�'� ��7�7�L� � � �|�4�4�Q�7�� '�#.�$�!3� � � � )�joblib�utilsr �loadr r r � r r �<module>r s] ��
�
�
�
� � � � � � ��� �0�1�1��
�V�[�/�
0�
0�5�6� 7� 8� 9� 10r