-
-
Notifications
You must be signed in to change notification settings - Fork 211
Closed
Description
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
Labels
No labels