def test_covariance(self):
start_time = time.time()
data = self.data.T
np_cov = np.cov(data)
logging.info("Numpy took %f", time.time() - start_time)
start_time = time.time()
with self.test_session() as sess:
op = gmm_ops._covariance(constant_op.constant(data.T, dtype=dtypes.float32), False)
op_diag = gmm_ops._covariance(constant_op.constant(data.T, dtype=dtypes.float32), True)
variables.global_variables_initializer().run()
tf_cov = sess.run(op)
np.testing.assert_array_almost_equal(np_cov, tf_cov)
logging.info("Tensorflow took %f", time.time() - start_time)
tf_cov = sess.run(op_diag)
np.testing.assert_array_almost_equal(np.diag(np_cov), np.ravel(tf_cov), decimal=5)
def test_covariance(self):
start_time = time.time()
data = self.data.T
np_cov = np.cov(data)
logging.info('Numpy took %f', time.time() - start_time)
start_time = time.time()
with self.test_session() as sess:
op = gmm_ops._covariance(
constant_op.constant(data.T, dtype=dtypes.float32), False)
op_diag = gmm_ops._covariance(
constant_op.constant(data.T, dtype=dtypes.float32), True)
variables.global_variables_initializer().run()
tf_cov = sess.run(op)
np.testing.assert_array_almost_equal(np_cov, tf_cov)
logging.info('Tensorflow took %f', time.time() - start_time)
tf_cov = sess.run(op_diag)
np.testing.assert_array_almost_equal(np.diag(np_cov),
np.ravel(tf_cov),
decimal=5)