TensorFlow函數(shù)：tf.metrics.mean_squared_error

tf.metrics.mean_squared_error函數(shù)

tf.metrics.mean_squared_error(
    labels,
    predictions,
    weights=None,
    metrics_collections=None,
    updates_collections=None,
    name=None
)

定義在：tensorflow/python/ops/metrics_impl.py.

計算labels和predictions之間的平均平方誤差.

該mean_squared_error函數(shù)創(chuàng)建兩個局部變量,total和count,它們被用來計算平均平方誤差.該平均值是通過weights加權(quán),并最終被返回為mean_squared_error：一種冪等運算,簡單地用count除total.

為了估計數(shù)據(jù)流上的度量,該函數(shù)創(chuàng)建一個update_op操作來更新這些變量并返回mean_squared_error.在內(nèi)部,squared_error操作計算predictions和labels之間差異的元素平方..然后update_op通過weights和squared_error乘積的減少總和來遞增total,并且它通過weights的減少總和來遞增count.

如果weights是None,則權(quán)重默認(rèn)為1,使用權(quán)重0來屏蔽值.

參數(shù)：

• labels：與predictions具有相同形狀的Tensor.
• predictions：任意形狀的Tensor.
• weights：可選的Tensor,其秩為0或與labels具有相同的秩,并且必須可廣播到labels(即,所有維度必須為1或者與相應(yīng)的labels維度相同).
• metrics_collections：mean_squared_error應(yīng)添加到的集合的可選列表.
• updates_collections：update_op應(yīng)添加到的集合的可選列表.
• name：可選的variable_scope名稱.

返回：

• mean_squared_error：表示當(dāng)前均值的Tensor,total除以count的值.
• update_op：適當(dāng)增加total和count變量,并且其值與mean_squared_error匹配的操作.

可能引發(fā)的異常：

• ValueError：如果predictions和labels有不匹配的形狀,或者weights不是None,并且它的形狀與predictions不匹配,或者如果metrics_collections或updates_collections中任意一個不是一個列表或元組.
• RuntimeError：如果啟用了急切執(zhí)行.