/NaiveBayesTAN

Consists of implementations of both Naive Bayes as well as TAN

Primary LanguagePython

NaiveBayesTAN

Consists of implementations of both Naive Bayes as well as TAN

Requirements

Python 2.7

Important Scripts

  • bayes: Script that calls bayes.py
  • bayes.py: Contains the implementation of Naive Bayes as well as TAN
  • *.arff: Sample data files

Usage

Command to run the script: ./bayes train-set-file test-set-file n|t

  • n: Naive Bayes
  • t: TAN

TAN algorithm uses Prim's algorithm to find the maximal spanning tree.