def testTrainBufferPerceptron(self): dataSetName = 'buffer_dataset' container = DataFetcher().fetchDataSet(dataSetName) perceptron = Perceptron(container, 0.1, 15) perceptron.train() weightVector = perceptron.getWeightVector() expectedLength = container.getDataVectorLength() + 1 self.assertEqual(len(weightVector), expectedLength)
def trainPerceptron(self, dataSetName, alpha, iterations): container = DataFetcher().fetchDataSet(dataSetName) perceptron = Perceptron(container, 0.1, 15) perceptron.train() return perceptron