Python RecognizerFlatten.fit Beispiele

Beispiel #1

Datei: test_new_recognizer.py Projekt: juyoungkorea/autofpop

class RecognizerFlattenTest(unittest.TestCase):
	def setUp(self):
		self.original_base = ImageReader.base
		ImageReader.base = 'tests/fixtures/' + ImageReader.base

		self.subject = RecognizerFlatten()

	def tearDown(self):
		ImageReader.base = self.original_base

	def test_fit_predict(self):
		self.subject.load_data(['BLACK', 'BLUE'])
		self.subject.fit()
		self.assertEqual(
			self.subject.predict(self.subject.data.X).shape[1], 3 * 50 * 50)
		for X, y in zip(self.subject.data.X, self.subject.data.y):
			self.assertEqual(self.subject.predict(X).shape, (3 * 50 * 50, ))

Beispiel #2

class RecognizerFlattenTest(unittest.TestCase):
    def setUp(self):
        self.original_base = ImageReader.base
        ImageReader.base = 'tests/fixtures/' + ImageReader.base

        self.subject = RecognizerFlatten()

    def tearDown(self):
        ImageReader.base = self.original_base

    def test_fit_predict(self):
        self.subject.load_data(['BLACK', 'BLUE'])
        self.subject.fit()
        self.assertEqual(
            self.subject.predict(self.subject.data.X).shape[1], 3 * 50 * 50)
        for X, y in zip(self.subject.data.X, self.subject.data.y):
            self.assertEqual(self.subject.predict(X).shape, (3 * 50 * 50, ))