8201900250bab24c768542f80090dc6381f05af1,tests/loss_regression_test.py,RMSELossTests,test_rmse_loss_value_weight,#RMSELossTests#,156

Before Change

    def test_rmse_loss_value_weight(self):
        with self.test_session():
            weights = tf.constant(
                [1, 2.1], dtype=tf.float32, name="weights")
            predicted = tf.constant(
                [1, 1], dtype=tf.float32, name="predicted")
            labels = tf.constant([1, 0], dtype=tf.float32, name="labels")
            test_loss_func = LossFunction(loss_type="RMSE")
            computed_rmse_loss = test_loss_func(predicted, labels, weights)
            self.assertAlmostEqual(
                computed_rmse_loss.eval(), 0.8231, places=4)

After Change

    def test_rmse_loss_value_weight(self):
        with self.test_session():
            weights = tf.constant(
                [[1, 2.1]], dtype=tf.float32, name="weights")
            predicted = tf.constant(
                [[1, 1]], dtype=tf.float32, name="predicted")
            labels = tf.constant([[1, 0]], dtype=tf.float32, name="labels")
            test_loss_func = LossFunction(loss_type="RMSE")
            computed_rmse_loss = test_loss_func(predicted, labels, weights)
            self.assertAlmostEqual(
                computed_rmse_loss.eval(), 0.8231, places=4)

Instances