search

Colin-Codes / IntentClassifier-ML-Project

Pyhton, Keras, SciKit-Learn, Matplotlib: Machine learning research project around classification of intent behind tech support emails in order to enable automatic follow up.
0 stars 0 forks source link

calculate Certainty of KNN predictions #33

Closed Colin-Codes closed 4 years ago

Colin-Codes commented 4 years ago

This is actually quite easy, using the model.predict_proba(X_test) method.

Colin-Codes commented 4 years ago
  1. Calculating certainty of KNN predictions: I thought this would be very difficult, as I assumed that the KNN model was only capable of giving classifications, but I have discovered that it can easily provide a probability score too, using the predict_proba method. However, I am also investigating normalising the results, to favour higher prediction probabilities.