class_docstr: Test Container usage.
docstr: Tests that an attention architecture can be created without crash.
docstr_tok: ["Tests","that","an","attention","architecture","can","be","created","without","crash","."]
code: def test_sample_attn_lstm_architecture(self):
      """Tests that an attention architecture can be created without crash."""
      max_depth = 5
      n_test = 5
      n_support = 11
      n_feat = 10
      batch_size = 3
  
      support_model = SequentialSupportGraph(n_feat)
  
      # Add layers
      support_model.add(dc.nn.GraphConv(64, n_feat, activation="relu"))
      # Need to add batch-norm separately to test/support due to differing
      # shapes.
      support_model.add_test(
          dc.nn.BatchNormalization(epsilon=1e-5, mode=1)
      )
      support_model.add_support(
          dc.nn.BatchNormalization(epsilon=1e-5, mode=1)
      )
      support_model.add(dc.nn.GraphPool())
  
      # Apply an attention lstm layer
      support_model.join(
          dc.nn.AttnLSTMEmbedding(n_test, n_support, 64, max_depth)
      )
  
      # Gather Projection
      support_model.add(dc.nn.Dense(128, 64))
      support_model.add_test(
          dc.nn.BatchNormalization(epsilon=1e-5, mode=1)
      )
      support_model.add_support(
          dc.nn.BatchNormalization(epsilon=1e-5, mode=1)
      )
      support_model.add(
          dc.nn.GraphGather(batch_size, activation="tanh")
      )
  
file_path: deepchem/contrib/one_shot_models/tests/test_graph_models.py
imports: from __future__ import print_function
  from __future__ import division
  from __future__ import unicode_literals
  import unittest
  import tensorflow
  import deepchem
  from tensorflow.python.framework import test_util
  from deepchem.models.tf_new_models.graph_models import SequentialGraph
  from deepchem.models.tf_new_models.graph_models import SequentialSupportGraph
label_desc: Use image node_id 4 for calling the TestGraphModels obj's underlying member method code with example usage: obj.test_sample_attn_lstm_architecture() without return types