gcamfer / gym-classification

Open AI gym environment for classification task

Hi 👋, I'm Guillermo Caminero Fernández

Data Scientist with a passion for electronics and cybersecurity

• 🔭 I’m currently interested on Computer vision and NLP
• 🌱 I’m currently learning PySpark, Tensorflow, OpenCV looking forward to apply the state of the art research in new projects.
• 🎮 I love electronics and embeded devices like arduino, raspberry-pi, TTGO,nvidia jetson nano...
• 🎷 I love music and I haven't practised for years but I have played saxophone and guitar.
• 🏀 I like sports, I've played basketball but I've been playing only on the couch for a long time, I hope this will change soon.

Connect with me:

Languages and Tools

IDEs:

Languages:

Develop:

ML:

GitHub Stats

gym-classification's Issues

IndexError: index 2373 is out of bounds for axis 0 with size 2373

Hello,
first thanks for sharing your code.
I run your code with MNIST dataset without any problem but when I use it for my dataset after 13 epochs ,I get an error on
# Update states for next step
self.true_labels = np.take(self.y,self.current_indices,axis=0).ravel()
and the erroe is:
IndexError: index 2373 is out of bounds for axis 0 with size 2373
do you know why this happend?

example MNIST does not work

Hello,
Thank you for your code. While I was running your MNIST example an error appeared.

I added statements in your code to prevent the error since the truth value of an array with more than one element is ambiguous.

But as a result, the model could not converge like that you showed in your github page.

I wonder if you had the same problem or is there something that I've missed.