Python make_protein_domain示例

示例#1

文件： embed_test.py 项目： zzhaozeng/google-research

 def setUp(self):
     self._domain = data.make_protein_domain(length=12)
     self._model = _get_model(self._domain)
     self._embed_fn = embed.get_embed_fn(model=self._model,
                                         domain=self._domain,
                                         reduce_fn=embed.masked_reduce_fn)
     super().setUp()

示例#2

文件： embed_test.py 项目： ziyouzizai111/google-research

  def test_encode_string_sequences(self, domain_length, length,
                                   expected_output_length):
    seqs = ['ABCD', 'EFG']
    domain = data.make_protein_domain(
        length=domain_length) if domain_length else None

    output_batch = embed._encode_string_sequences(
        seqs, domain=domain, length=length)
    self.assertEqual(output_batch.shape, (2, expected_output_length))

示例#3

文件： embed_test.py 项目： ziyouzizai111/google-research

 def setUp(self):
   self._domain = data.make_protein_domain(length=12)
   self._model = _get_model(self._domain)
   super().setUp()