Skip to content

Instantly share code, notes, and snippets.

@vivek1240

vivek1240/support_email_classification_medium_1.4.py

Created May 30, 2021 10:03
Show Gist options
  • Save vivek1240/0f913746037764aaaffd90afff0a41df to your computer and use it in GitHub Desktop.
Save vivek1240/0f913746037764aaaffd90afff0a41df to your computer and use it in GitHub Desktop.
Download ZIP
Raw
support_email_classification_medium_1.4.py
#Making Predictions
prediction_category= {0:'Refunds',1:'Cancellation',2:'Others',3:'Amendment',4:'Website Error'}
my_prediction = clf2.predict([data])
my_prediction= my_prediction[0]
my_prediction= prediction_category[my_prediction]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment