pygip.models.defense.atom.ATOM
==============================

.. automodule:: pygip.models.defense.atom.ATOM

   
   
   

   
   
   .. rubric:: Functions

   .. autosummary::
   
      average_pooling_with_neighbors
      average_pooling_with_neighbors_batch
      build_loaders
      collate_fn_no_pad
      compute_embedding_batch
      compute_returns_and_advantages
      custom_reward_function
      get_node_embedding
      get_one_hop_neighbors
      k_core_decomposition
      load_data_and_model
      precompute_all_node_embeddings
      precompute_simple_embeddings
      preprocess_sequences
      set_seed
      simple_embedding_batch
      split_and_adjust
      test_model
      train_gcn
   
   

   
   
   .. rubric:: Classes

   .. autosummary::
   
      ATOM
      FusionGRU
      GCN
      Memory
      PPOAgent
      PolicyNetwork
      SequencesDataset
      StateTransformMLP
      TargetGCN