cost.py 文件源码

python
阅读 30 收藏 0 点赞 0 评论 0

项目:dtnn 作者: atomistic-machine-learning 项目源码 文件源码
def calc_errors(self, output):
        Z = output['numbers']
        N = tf.reduce_sum(tf.cast(tf.greater(Z, 0), tf.float32), 1)

        tgt = output[self.target]
        pred = output[self.prediction]
        if self.idx is not None:
            tgt = tgt[:, self.idx]
            pred = pred[:, self.idx]
        return ((tgt - pred) / N) ** 2
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号