Ejemplos de build_model en Python

Ejemplo n.º 1

Archivo: visualization_test.py Proyecto: qixiuai/tensor2tensor

  def test_build_model_beam(self):
    inputs, targets, outputs, _ = visualization.build_model(
        hparams_set, model_name, self.data_dir, problem_name, beam_size=8)

    self.assertAllEqual((1, None, 1, 1), inputs.shape.as_list())
    self.assertAllEqual((1, None, 1, 1), targets.shape.as_list())
    self.assertAllEqual((None, None), outputs.shape.as_list())

Ejemplo n.º 2

Archivo: visualization_test.py Proyecto: xingjinglu/tensor2tensor

  def test_build_model_beam(self):
    inputs, targets, outputs, _ = visualization.build_model(
        hparams_set, model_name, self.data_dir, problem_name, beam_size=8)

    self.assertAllEqual((1, None, 1, 1), inputs.shape.as_list())
    self.assertAllEqual((1, None, 1, 1), targets.shape.as_list())
    self.assertAllEqual((None, None), outputs.shape.as_list())