svae
svae copied to clipboard
small diffs
I saw you fixed the from test_util (change to svae.util), but somehow it reverted on trunk? perhaps flesh out the setup.py a bit. Here is what I did (version requirements are just whatever I had available on my Ubuntu 16.04 box). I did not see the other experiments/ that wuaalb was mentioning. Perhaps they are on a branch?
diff --git a/setup.py b/setup.py
index c841313..01b893c 100644
--- a/setup.py
+++ b/setup.py
@@ -3,6 +3,16 @@ import numpy as np
from Cython.Build import cythonize
setup(
+ name='svae',
+ version='0.0.0',
+ description='structure variational auto-encoder',
+ install_requires=['autograd>=1.1.7', 'numpy>=1.11.0', 'scipy>=0.17.0', 'Cython>=0.25.1'
+ , 'pyhsmm>=0.1.6', 'toolz>=0.8.1'],
+ keywords=['autoencoder', 'machine learning', 'optimization'
+ , 'neural networks', 'Python', 'Numpy', 'Scipy'],
+ url='https://github.com/mattjj/svae',
+ packages=['svae', 'svae.distributions', 'svae.hmm', 'svae.lds', 'svae.models'],
+
ext_modules=cythonize('**/*.pyx'),
include_dirs=[np.get_include(),],
)
diff --git a/tests/test_gaussian.py b/tests/test_gaussian.py
index 7b7e560..2fa0daa 100644
--- a/tests/test_gaussian.py
+++ b/tests/test_gaussian.py
@@ -5,8 +5,7 @@ from autograd import grad
from svae.distributions.gaussian import logZ, expectedstats, \
pack_dense, unpack_dense
-from test_util import rand_psd
-
+from svae.util import rand_psd
def rand_gaussian(n):
J = rand_psd(n) + n * np.eye(n)