mindsponge.metrics.supervised_chi

查看源文件
mindsponge.metrics.supervised_chi(sequence_mask, aatype, sin_cos_true_chi, torsion_angle_mask, sin_cos_pred_chi, sin_cos_unnormalized_pred, chi_weight, angle_norm_weight, chi_pi_periodic)[源代码]

计算主链与侧链扭转角的误差,扭转角用角度的正弦与余弦值表示,该误差由两项组成,第一项是正则化后的预测正弦余弦值与真实值的角度差,第二项是预测值的模量与1的差值,称为角度模量误差。 Jumper et al. (2021) Suppl. Alg. 27 “torsionAngleLoss”

参数:
  • sequence_mask (Tensor) - 序列残基的mask,shape为 \((N_{res},)\) ,其中 \(N_{res}\) 是蛋白质中的残基数目。

  • aatype (Tensor) - 序列中的氨基酸残基类型,shape为 \((N_{res},)\)

  • sin_cos_true_chi (Tensor) - shape为 \((N_{res}, 14)\) ,扭转角的正弦和余弦值,每个氨基酸残基有七个扭转角,其中主链三个,侧链四个。

  • torsion_angle_mask (Tensor) - 侧链扭转角的mask,shape为 \((N_{res}, 4)\)

  • sin_cos_pred_chi (Tensor) - shape为 \((N_{res}, 4, 2)\) ,预测的侧链扭转角的正弦和余弦值。

  • sin_cos_unnormalized_pred (Tensor) - 预测的扭转角的正弦和余弦值,没有做过正则化,shape为 \((N_{recycle}, N_{res}, 7, 2)\) ,其中 \(N_{recycle}\) 是Structure模块中的循环次数。

  • chi_weight (float) - 角度差损失函数项的权重。

  • angle_norm_weight (float) - 角度模量损失函数项的权重。

  • chi_pi_periodic (Tensor) - 扭转角的周期性信息,某些氨基酸的某些扭转角具有周期性。氨基酸性质的常量,shape是 \((21, 4)\) ,21代表二十种氨基酸加未知氨基酸。

返回:

Tensor,主链与侧链扭转角的误差,shape为 \(()\)

支持平台:

Ascend GPU

样例:

>>> import numpy as np
>>> np.random.seed(0)
>>> from mindsponge.metrics import supervised_chi
>>> from mindsponge.common import residue_constants
>>> from mindspore import dtype as mstype
>>> from mindspore import Tensor
>>> sequence_mask = Tensor(np.random.rand(256, )).astype(mstype.float32)
>>> aatype = Tensor(np.random.randint(0, 21, (256,) )).astype(mstype.int32)
>>> sin_cos_true_chi = Tensor(np.random.rand(256, 4, 2)).astype(mstype.float32)
>>> torsion_angle_mask = Tensor(np.random.rand(256, 4)).astype(mstype.float32)
>>> sin_cos_pred_chi = Tensor(np.random.rand(256, 14)).astype(mstype.float32)
>>> sin_cos_unnormalized_pred = Tensor(np.random.rand(8, 256, 7, 2)).astype(mstype.float32)
>>> chi_weight = 0.1
>>> angle_norm_weight = 0.2
>>> chi_pi_periodic = Tensor(residue_constants.chi_pi_periodic).astype(mstype.float32)
>>> chi_loss = supervised_chi(sequence_mask, aatype, sin_cos_true_chi, torsion_angle_mask, sin_cos_pred_chi,
...                           sin_cos_unnormalized_pred, chi_weight, angle_norm_weight, chi_pi_periodic)
>>> print(chi_loss)
0.061829045