Ejemplo n.º 1
0
	def maxpoolTest(self):
		nn = NeuralNet()

		# Should fail - Wrong size array
		expected = np.array([[4,4], [0,4]])

		data = np.array([[4,0,1,3,4], [0,0,2,4,4], [0,0,4,4,4], [0,0,4,4,4]])
		windowsize = 2 # 2x2 window

		output = nn.maxPool(data, windowsize)

		if output is not None:
			return False

		# Should pass
		expected = np.array([[4,4], [0,4]])

		data = np.array([[4,0,1,3], [0,0,2,4], [0,0,4,4], [0,0,4,4]])
		windowsize = 2 # 2x2 window

		output = nn.maxPool(data, windowsize)

		return self.arrayIsSame(output,expected)