AMROUN Lysa I believe you are looking to create a model wihich is ONNX compatible with AutoML as per your previous thread. In this case in the automl config if you setup enable_onnx_compatible_models
to true in the automl config the model should be available for download. Something similar to what is provided as guidance in this notebook.
I would recommend running through this notebook and then change your experiment settings in a similar way to retrieve the best onnx format model.
automl_settings = {
"experiment_timeout_hours": 0.3,
"enable_early_stopping": True,
"iteration_timeout_minutes": 5,
"max_concurrent_iterations": 4,
"max_cores_per_iteration": -1,
# "n_cross_validations": 2,
"primary_metric": "AUC_weighted",
"featurization": "auto",
"verbosity": logging.INFO,
"enable_code_generation": True,
}
automl_config = AutoMLConfig(
task="classification",
debug_log="automl_errors.log",
compute_target=compute_target,
experiment_exit_score=0.9984,
blocked_models=["KNN", "LinearSVM"],
enable_onnx_compatible_models=True,
training_data=train_data,
label_column_name=label,
validation_data=validation_dataset,
**automl_settings,
)
If this answers your query, do click Accept Answer
and Yes
for was this answer helpful. And, if you have any further query do let us know.