Python AutoEncoder.get_variables_to_initの例

コード例 #1

ファイル: autoencoder_test.py プロジェクト: Lelouch5023/TensorFlowDeepAutoencoder

  def test_get_variables(self):
    with self.test_session() as sess:
      ae_shape = [10, 20, 30, 2]
      ae = AutoEncoder(ae_shape, sess)

      with self.assertRaises(AssertionError):
        ae.get_variables_to_init(0)
      with self.assertRaises(AssertionError):
        ae.get_variables_to_init(4)

      v1 = ae.get_variables_to_init(1)
      self.assertEqual(len(v1), 3)

      v2 = ae.get_variables_to_init(2)
      self.assertEqual(len(v2), 5)

      v3 = ae.get_variables_to_init(3)
      self.assertEqual(len(v3), 2)

コード例 #2

ファイル: autoencoder_test.py プロジェクト: auroua/sar_edge_detection

    def test_get_variables(self):
        with self.test_session() as sess:
            ae_shape = [10, 20, 30, 2]
            ae = AutoEncoder(ae_shape, sess)

            with self.assertRaises(AssertionError):
                ae.get_variables_to_init(0)
            with self.assertRaises(AssertionError):
                ae.get_variables_to_init(4)

            v1 = ae.get_variables_to_init(1)
            self.assertEqual(len(v1), 3)

            v2 = ae.get_variables_to_init(2)
            self.assertEqual(len(v2), 5)

            v3 = ae.get_variables_to_init(3)
            self.assertEqual(len(v3), 2)