mindsponge.metrics.compute_renamed_ground_truth

查看源文件
mindsponge.metrics.compute_renamed_ground_truth(atom14_gt_positions, atom14_alt_gt_positions, atom14_atom_is_ambiguous, atom14_gt_exists, atom14_pred_positions, atom14_alt_gt_exists)[源代码]

由于蛋白质中部分氨基酸侧链存在对称性,因此可能存在等价的构象。本函数根据预测的蛋白质三维坐标与多种等价的构象作比较,并从中选出和预测坐标最接近的构象坐标作为结构标签。

参数:
  • atom14_gt_positions (Tensor) - 按照稠密编码方式编码,蛋白质全原子坐标。shape \((N_{res}, 14, 3)\)

  • atom14_alt_gt_positions (Tensor) - 按稠密编码方式编码,根据对称性变换的等价全原子坐标。shape \((N_{res}, 14, 3)\)

  • atom14_atom_is_ambiguous (Tensor) - 由于部分氨基酸结构具有局部对称性,其对称原子编码可调换,具体原子参考 common.residue_atom_renaming_swaps 该特征记录了原子不确定的编码位置。shape \((N_{res}, 14)\)

  • atom14_gt_exists (Tensor) - 按照稠密编码方式编码,表示蛋白的相应原子在标签中是否存在的掩码。shape \((N_{res}, 14)\)

  • atom14_pred_positions (Tensor) - 按照稠密编码方式编码,预测的蛋白质全原子坐标。shape \((N_{res}, 14, 3)\)

  • atom14_alt_gt_exists (Tensor) - 按照稠密编码方式编码,表示蛋白的等价构象中的相应原子在标签中是否存在的掩码。shape \((N_{res}, 14)\)

返回:
  • alt_naming_is_better (Tensor) - 记录了所有氨基酸是否在对称变换后更贴近预测坐标。shape \((N_{res}, )\)

  • renamed_atom14_gt_positions (Tensor) - 对称变换后重命名的,稠密编码的原子所对应的真实三维坐标。shape \((N_{res}, 14, 3)\)

  • renamed_atom14_gt_exists (Tensor) - 对称变换后重命名的、稠密编码的原子在标签中是否存在的掩码。shape \((N_{res}, 14)\)

符号:

\(N_{res}\) - 蛋白质中氨基酸个数,按蛋白质一级序列排列。

支持平台:

Ascend GPU

样例:

>>> import mindspore as ms
>>> from mindspore import Tensor
>>> import numpy as np
>>> from mindsponge.metrics import compute_renamed_ground_truth
>>> atom14_gt_positions = Tensor(np.random.random(size=(50, 14, 3)), ms.float32)
>>> atom14_alt_gt_positions = Tensor(np.random.random(size=(50, 14, 3)), ms.float32)
>>> atom14_atom_is_ambiguous = Tensor(np.random.random(size=(50, 14)), ms.float32)
>>> atom14_gt_exists = Tensor(np.random.random(size=(50, 14)), ms.float32)
>>> atom14_pred_positions = Tensor(np.random.random(size=(50, 14, 3)), ms.float32)
>>> atom14_alt_gt_exists = Tensor(np.random.random(size=(50, 14)), ms.float32)
>>> alt_naming_is_better, renamed_atom14_gt_positions, renamed_atom14_gt_exists =         ...     compute_renamed_ground_truth(atom14_gt_positions, atom14_alt_gt_positions, atom14_atom_is_ambiguous,
...                                  atom14_gt_exists, atom14_pred_positions, atom14_alt_gt_exists)
>>> print(alt_naming_is_better.shape, renamed_atom14_gt_positions.shape, renamed_atom14_gt_exists.shape)
(50,) (50, 14, 3) (50, 14)