def __init__(self, trainingDataPath, salaryProbabilityPath, distributionGranularity,
dataGranularity, grammar, dropOffWidth = 0, stepDown = 0,
increment = 5.0):
self.salaryData = SalaryDistribution(salaryProbabilityPath, distributionGranularity)
self.salaryData.parse()
self.trainingDataPath = trainingDataPath
self.dataGranularity = dataGranularity
self.stochasticGrammar = grammar
#print self.stochasticGrammar.wordData['surrey'].values[8000]
#print self.stochasticGrammar.wordData['limited'].values[8000]
#print self.stochasticGrammar.wordData['experience'].values[8000]
#self.stochasticGrammar.wordData['surrey'].increment(8250, 0, 0)
#self.stochasticGrammar.wordData['experience'].increment(8250, 0, 0)
#print self.stochasticGrammar.wordData['surrey'].values[8000]
#print self.stochasticGrammar.wordData['limited'].values[8000]
#print self.stochasticGrammar.wordData['experience'].values[8000]
self.dropOffWidth = dropOffWidth
self.stepDown = stepDown
self.increment = increment
dummy = Word("", self.dataGranularity)
dummy.initializeNewValues(self.salaryData)
self.initialValue = dummy.values
def wordTest(inputPath, distributionGranularity, wordGranularity, tempFile):
salaryDictionary = SalaryDistribution(inputPath, distributionGranularity)
salaryDictionary.parse()
test1 = Word("Potato", wordGranularity)
test1.initializeNewValues(salaryDictionary)
test2 = Word("Celery", wordGranularity)
test2.initializeNewValues(salaryDictionary)
with open(tempFile, 'w') as dataDump:
dataDump.write(test1.configure())
dataDump.write(test2.configure())
with open(tempFile, 'r') as reReading:
wordList = []
for line in reReading:
try:
wordList.append(Word.fromFileString(line))
except NameError:
continue
print sum(wordList[0].values.values())
print sum(wordList[1].values.values())
wordList[0].increment(8250,increment = 147.25)
print sum(wordList[0].values.values())
print sum(wordList[1].values.values())
