mindsponge.metrics.backbone

查看源文件
mindsponge.metrics.backbone(traj, backbone_affine_tensor, backbone_affine_mask, fape_clamp_distance, fape_loss_unit_distance, use_clamped_fape)[源代码]

调用 frame_aligned_point_error_map 实现骨架全原子损失函数计算 Jumper et al. (2021) Suppl. Alg. 20 “StructureModule” line 17.

参数:
  • traj (Tensor) - Structure模块预测的一系列骨架局部坐标系(轨迹),shape为 \((N_{recycle}, N_{res}, 7)\) ,其中 \(N_{recycle}\) 是Structure模块中的循环次数, \(N_{res}\) 是蛋白质中的残基数目,对于最后一维,前四个分量是表征旋转的四元数,代表局部坐标系相对全局坐标系的旋转,后三个分量是三维空间的平移。

  • backbone_affine_tensor (Tensor) - 真实的的骨架局部坐标系,shape为 \((N_{res}, 7)\)

  • backbone_affine_mask (Tensor) - 骨架局部坐标系的mask,shape为 \((N_{res},)\)

  • fape_loss_unit_distance (float) - 单位距离,用于缩放距离的差,常量。

  • fape_clamp_distance (float) - 距离误差的截断点,超过该距离时梯度不再考虑,常量。

  • use_clamped_fape (float) - 是否截断截断距离误差, 0 或者 10 代表不截断。

返回:
  • fape (list) - Tensor。计算所得Structure模块最后一次迭代输出的结构的全原子点位置误差,如果use_clamped_fape为1,则计算过程中过大的误差会被截断。shape为 \(()\)

  • loss (list) - Tensor。计算所得Structure模块所有迭代输出的结构的全原子点位置误差的均值,如果use_clamped_fape为1,则计算过程中过大的误差会被截断。shape为 \(()\)

  • no_clamp (list) - Tensor。计算所得Structure模块最后一次迭代输出的结构的全原子点位置误差,没有截断。shape为 \(()\)

支持平台:

Ascend GPU

样例:

>>> import numpy as np
>>> np.random.seed(0)
>>> from mindsponge.metrics import backbone
>>> from mindspore import dtype as mstype
>>> from mindspore import Tensor
>>> traj = Tensor(np.random.rand(8, 256, 7)).astype(mstype.float32)
>>> backbone_affine_tensor = Tensor(np.random.rand(256, 7)).astype(mstype.float32)
>>> backbone_affine_mask = Tensor(np.random.rand(256,)).astype(mstype.float16)
>>> fape_clamp_distance = 10.0
>>> fape_loss_unit_distance = 10.0
>>> use_clamped_fape = 1
>>> fape, loss, noclamp = backbone(traj, backbone_affine_tensor, backbone_affine_mask,
...                                fape_clamp_distance, fape_loss_unit_distance, use_clamped_fape)
>>> print(fape, loss, noclamp)
0.12813742 0.12904957 0.12813742