The examples in the docstrings for SKLearnClassifier
and SKLearnRegressor
are currently not working. This is because of several small issues:
- General issues:
- Importing layers from keras.src.layers
instead of keras.layers
- Importing Model from keras.src.layers
instead of keras.models
- Unnecessary usage of []
around the output layer
- SKLearnClassifier
docstring issue:
- The example uses n_classes=3
, but this leads to a scikit-learn error: ValueError: n_classes(3) * n_clusters_per_class(2) must be smaller or equal 2**n_informative(2)=4
- SKLearnRegressor
docstring issue:
- Uses softmax
activation where linear
would be more appropriate (regression problem that does not fall between [0, 1]
), causing warnings.
I will open a PR to address these issues shortly.