Skip to content

KerasClassifier : No extension registered which can handle model #1067

@sahithyaravi

Description

@sahithyaravi

Description

No extension registered error when trying to use keraswrapper
The lastest flow with Keraswrapper/KerasClassifier seems to be based on openml v0.8.0

Steps/Code to Reproduce

Example:

from keras.wrappers.scikit_learn import KerasClassifier
task = openml.tasks.get_task(3573)  # mnist
model = KerasClassifier(build_fn=your_model_build_function, epochs=8, batch_size=32, verbose=1)
openml.runs.run_model_on_task(model, task)

Expected Results

No error is thrown.

Actual Results

ValueError                                Traceback (most recent call last)
<ipython-input-6-c3c90f60e704> in <module>()
      2 mnist_task=openml.tasks.get_task(3573)
      3 # Run the Keras model on the task (requires an API key).
----> 4 mnist_lenet_run =openml.runs.run_model_on_task(mnist_task, lenet_model, dataset_format ="array")
      5 # Publish the experiment on OpenML (optional, requires an API key).
      6 #myrun = mnist_lenet_run.publish()

1 frames
/usr/local/lib/python3.7/dist-packages/openml/extensions/functions.py in get_extension_by_model(model, raise_if_no_extension)
     90     if len(candidates) == 0:
     91         if raise_if_no_extension:
---> 92             raise ValueError("No extension registered which can handle model: {}".format(model))
     93         else:
     94             return None

ValueError: No extension registered which can handle model: <tensorflow.python.keras.wrappers.scikit_learn.KerasClassifier object at 0x7f5bfdf6e2d0>

Versions

Linux-4.19.112+-x86_64-with-Ubuntu-18.04-bionic
Python 3.7.10 (default, Feb 20 2021, 21:17:23)
[GCC 7.5.0]
NumPy 1.19.5
SciPy 1.4.1
Scikit-Learn 0.22.2.post1
OpenML 0.12.1

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions