naive-bayes-classifier-/code_for_classification.py at master · akshat2412/naive-bayes-classifier- · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from sys import argv
from math import log
import pymysql

cursor = ""
class_prob={}
prob={}

def global_variables():
	global class_prob
	global prob
	global cursor
	db = pymysql.connect("localhost","root","root","naive" )
	cursor = db.cursor()
	sql = "select * from probability_class"
	cursor.execute(sql)
	resultSet=cursor.fetchall()
	for row in resultSet:
		class_prob[row[0]]=row[1]

	# print('How many times will there statments over execute ? ................... test #26')

	sql = "select * from probability_word_given_class"
	cursor.execute(sql)
	resultSet=cursor.fetchall()
	for row in resultSet:
		if row[1] not in prob:
			prob[row[1]]={}
		prob[row[1]][row[0]]=row[2]

	db.close()


def read_words(words_file):
    return [word for line in open(words_file, 'r') for word in line.split()]


def findClass(target):
	global_variables()
	# target = read_words(target)
	max_prob = float("-inf")
	max_class = ""
	for _class in class_prob:
		_prob = log(class_prob[_class])
		for _word in target:
			if _word in prob:
				_prob += log(prob[_word][_class])
		if _prob>max_prob:
			max_prob = _prob
			max_class = _class

	return max_class

if __name__=="__main__":
	# script,filename = argv
	filename="test_input.txt"
	text = read_words(filename)
	global_variables()
	print(findClass(text))