def test_untangent():
covmat = CovMat.random(10)
covmats = CovMats.random(10, 10)
tangent = TangentSpace.tangent(covmats, covmat)
old = untangent_space(tangent, covmat.numpy_array)
covmats.reset_covmats_fields()
new = TangentSpace.untangent(tangent, covmat)
return _get_state(old, new, "untangent")
def test_untangent():
covmat = CovMat.random(10)
covmats = CovMats.random(10, 10)
tangent = TangentSpace.tangent(covmats, covmat)
old = untangent_space(tangent, covmat.numpy_array)
covmats.reset_covmats_fields()
new = TangentSpace.untangent(tangent, covmat)
return _get_state(old, new, "untangent")
def test_tangent():
covmat = CovMat.random(10)
covmats = CovMats.random(10, 10)
old = tangent_space(covmats.numpy_array, covmat.numpy_array)
covmats.reset_covmats_fields()
new = TangentSpace.tangent(covmats, covmat)
return _get_state(old, new, "tangent space")
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..")))
from Utils.CovMat import CovMat
from Utils.CovMats import CovMats
from Utils.TangentSpace import TangentSpace
size = [10, 25, 50, 75, 100, 250, 500, 750, 1000]
# WARMUP
print("Warm up...")
for i in range(0, 10):
warm_up_covmat = CovMat.random(1000)
warm_up_covmat.expm
for i in range(0, len(size)):
covmats = CovMats.random(10, size[i])
covmat = CovMat.random(size[i])
tangent = TangentSpace.tangent(covmats, covmat)
t = timeit.Timer("untangent_space(tangent, covmat.numpy_array)",
setup="from __main__ import tangent, covmat; from oldPyRiemann.tangentspace import untangent_space; import Utils.OpenBLAS")
old_time = t.timeit(number=size[len(size) - i - 1]) / size[len(size) - i - 1]
t = timeit.Timer("covmat.reset_fields(); TangentSpace.untangent(tangent, covmat)",
setup="from Utils.TangentSpace import TangentSpace; from __main__ import tangent, covmat")
new_time = t.timeit(number=size[len(size) - i - 1]) / size[len(size) - i - 1]
print("matrix size : " + "10x" + str(size[i]) + "x" + str(size[i]) + "\t\told time : " + str(
old_time) + " sec\t\t" + "new time : " + str(new_time) + " sec\t\t" + "speed up : " + str(
old_time / new_time))
