__author__ = 'Aman'
import DataPrepare
import DataSpecific
import DecisionTree
'''
tree is the root of DecisionTree obtained by training the data in data file
'''
tree = DecisionTree.trainData()
'''
Updating the attribute list with values obtained from converting continuous variables.
'''
DataSpecific.attribute_list.update(tree.dictIntervalContVar)
'''
Convert continuous variables from the query/test data to discrete variables using the intervals used in decision tree.
'''
def cleanRow(row):
row = DataPrepare.cleanRow(row)
dictContVar = tree.dictIntervalContVar
for entry in dictContVar.keys():
listIntervals = dictContVar[entry]
for i in range(0,len(listIntervals)-1):
if row[entry] == '?':
break
if row[entry] > listIntervals[i] and row[entry] <= listIntervals[i+1]:
row[entry] = i
break
