example MNIST does not work about gym-classification HOT 2 OPEN

Hi, thank you very much for trying it and finding issues.
I don't really know why I returned a list of booleans, I guess my head was thinking about other stuff.
I have fixed that problem.

Regarding the convergence, it seems to be a hyperparameter problem. Try reducing the exploration or reducing the learning rate. You can even rerun as random initialisation will get you to another point, but it's not very stable.

You can try the branch in this Colab and next I'll merge with master.

Greetings

from gym-classification.

Hello,
After execution of the code I am getting this error

UndefinedMetricWarning: Precision and F-score are ill-defined and being set to 0.0 in labels with no predicted samples. Use the zero_division parameter to control this behaviour.

this error is resulting in no f1 score for some labels.
can you give me any solution for this error

from gym-classification.