eamid / examples Goto Github PK

Good Morning. I tried the example code in Window, using Pycharm, anaconda virt env of python=3.8, cuda=11.3.1, with PyTorch.
First, in your code, trimap called the function name 'trasform()' to clustering while others uses fit_transform.
When I run the code, AttributeError: module 'trimap' has no attribute 'transform' has alarmed.
I edited this line into trimap.TRIMAP().fit_transform(key, examples) but then the error said that key = random.PRNGKey(123) should be in 2-dimensional but only returns 1-dimensional so the 'n, dim = X.shape()' could be operated properly.
I searched for the jax.random.PRNGKey() but all the examples uses it in additional step to make actual random number.
I installed jax and jaxlib in pip in anaconda, jax with sourcecode and jaxlib with wheel.
If there's any advice to offer, thanks in advance.

HI,
Thank you for this example notebook. It's amazing. Could you please share requirements.txt as I am facing some version issues with numpy.