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
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions