HF RL Explorer

Task: Fix StratifiedKFold ValueError with Multi-class One-Hot Encoded Labels

Task: Fix StratifiedKFold ValueError with Multi-class One-Hot Encoded Labels: a task in Terminal-Lego-15k (Harbor dataset). When using scikit-learn's StratifiedKFold with one-hot encoded (categorical) labels produced by keras.utils.to categorical() , a ValueError is raised:

The task

When using scikit-learn's `StratifiedKFold` with one-hot encoded (categorical) labels produced by `keras.utils.to_categorical()`, a `ValueError` is raised:

Part of PrimeIntellect/Terminal-Lego-15k.