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())