Skip to content

swap357/python-naive-bayes-classifier

Repository files navigation

python-naive-bayes-classifier

Files

train_classifier.py
  • Trains the NaiveBayesClassifier, dumps the classifier into classifier.pickle for use
classifier.pickle
  • Contains the classifier object prepared by train_classifier.py
features.py
  • Contains the features/factors to consider for classification.
use_classifier.py
  • Once the classifier is ready, use the classifier to classify new data.
male.txt and female.txt
  • Provides sample dataset to train your classifier, you replace and use yours.