ai2_kit.algorithm.uninmr package

Subpackages

• ai2_kit.algorithm.uninmr.data package
  • Submodules
    • ai2_kit.algorithm.uninmr.data.conformer_dataset module
    • ai2_kit.algorithm.uninmr.data.cropping_dataset module
    • ai2_kit.algorithm.uninmr.data.distance_dataset module
    • ai2_kit.algorithm.uninmr.data.key_dataset module
    • ai2_kit.algorithm.uninmr.data.lattice_dataset module
    • ai2_kit.algorithm.uninmr.data.lmdb_dataset module
    • ai2_kit.algorithm.uninmr.data.mask_points_chunk_dataset module
    • ai2_kit.algorithm.uninmr.data.mask_points_dataset module
    • ai2_kit.algorithm.uninmr.data.normalize_dataset module
    • ai2_kit.algorithm.uninmr.data.pad_dataset module
    • ai2_kit.algorithm.uninmr.data.remove_hydrogen_dataset module
    • ai2_kit.algorithm.uninmr.data.resample_dataset module
    • ai2_kit.algorithm.uninmr.data.select_token_dataset module
  • Module contents
• ai2_kit.algorithm.uninmr.losses package
  • Submodules
    • ai2_kit.algorithm.uninmr.losses.atom_regloss module
    • ai2_kit.algorithm.uninmr.losses.regloss module
    • ai2_kit.algorithm.uninmr.losses.unimat module
    • ai2_kit.algorithm.uninmr.losses.unimat_rcut module
  • Module contents
• ai2_kit.algorithm.uninmr.models package
  • Submodules
    • ai2_kit.algorithm.uninmr.models.transformer_encoder_with_pair module
    • ai2_kit.algorithm.uninmr.models.unimat module
  • Module contents
• ai2_kit.algorithm.uninmr.tasks package
  • Submodules
    • ai2_kit.algorithm.uninmr.tasks.unimat module
    • ai2_kit.algorithm.uninmr.tasks.unimat_rcut module
    • ai2_kit.algorithm.uninmr.tasks.uninmr module
  • Module contents
• ai2_kit.algorithm.uninmr.utils package
  • Subpackages
    • ai2_kit.algorithm.uninmr.utils.data_preparation package
      • Submodules
      • Module contents
  • Submodules
    • ai2_kit.algorithm.uninmr.utils.data_scaler module
    • ai2_kit.algorithm.uninmr.utils.get_result module
    • ai2_kit.algorithm.uninmr.utils.parse module
  • Module contents

Submodules

• ai2_kit.algorithm.uninmr.infer module

Module contents