e2f51920215f2c3be39de713626d1288c607eb75,test/test_auto_encoder.py,,test_rank1_auto_encoder,#,6

Before Change

        for j in range(XY.shape[1]):
            expected = XY[i, j]
            predicted = XY_completed[i, j]
            assert abs(expected - predicted) < 0.1, \
                "Expected %0.4f but got %0.4f at XY[%d,%d]" % (
                    expected,
                    predicted,
                    i,
                    j)


if __name__ == "__main__":
    test_rank1_auto_encoder()

After Change

    print(XY)
    print(XY_completed)
    print(missing_mask)
    diff = XY - XY_completed
    missing_mse = np.mean(diff[missing_mask] ** 2)
    missing_mae = np.mean(np.abs(diff[missing_mask]))
    print("MSE: %0.4f, MAE: %0.4f" % (missing_mse, missing_mae))
    assert missing_mae < 0.1, "Error too high!"

Instances