def classify(self, trainedObj, datasetFilename,resultsFilename): dataset = inputProcessor.processFile(datasetFilename,False) trueRowValues = trainedObj.classify(dataset,0) inputProcessor.writeToFile(trueRowValues,resultsFilename)
def train(self, trainingFilename): input = inputProcessor.processFile(trainingFilename,True) self.learner = DecisionTreeLearner(input) return self.learner