Karthikbt/Emotion_Classification_Model
0
1a
2 #ChU � @ s� d dl Zd dlmZ d dlmZ d dlmZmZm Z d dl3m Z dZ e e �\Z
ZZZe� Ze�e
�Ze�e�Zedd�Ze�ee� e�e�Zed eee�� ed4eee�� d d d
dddd�Zede eedd� e d�D �d�� dS )� N)�TfidfVectorizer)�LogisticRegression)�accuracy_score�confusion_matrix�classification_report)�load_and_preprocesszdata/emotion_dataset.csvi� )Zmax_iterz Accuracy:zConfusion Matrix:5Zsadness�joyZlove�angerZfearZsurprise)r � � � � � zClassification Report:6c C s g | ] }t | �qS � )� label_map)�.0�ir r �E/Users/karthikbt/Desktop/AI-ML/emotion-classifier/train_classifier.py�7<listcomp>$ � r � )Z target_names) �pandas�pdZ sklearn.feature_extraction.textr Zsklearn.linear_modelr Zsklearn.metricsr r r �data.preprocessr Zcsv_pathZX_trainZX_testZy_trainZy_test�8vectorizerZ
fit_transformZ X_train_vec� transformZ9X_test_vec�clf�fit�predictZy_pred�printr �ranger r r r �<module> s0 101112 13��