nirs4all.pipeline.dagml.autogluon_estimator module

Cloneable DAG host adapter for AutoGluon’s directory-backed predictor.

class nirs4all.pipeline.dagml.autogluon_estimator.DagMLAutoGluonEstimator(params: dict[str, Any] | None = None, fit_params: dict[str, Any] | None = None)[source]

Bases: BaseEstimator

Fit one AutoGluon predictor per DAG fold and keep its artifact replayable.

fit(X: Any, y: Any) → DagMLAutoGluonEstimator[source]

load_external_directory(directory: Path) → None[source]: Load a verified sidecar after the enclosing archive has been checked.

predict(X: Any) → ndarray[source]

predict_proba(X: Any) → ndarray[source]

nirs4all.pipeline.dagml.autogluon_estimator.autogluon_step_estimator(step: dict[str, Any]) → DagMLAutoGluonEstimator | None[source]

Adapt the explicit configurations routed to the legacy AutoGluon controller.

Legacy reads constructor/fit options from the step’s top-level params and applies model_params as overrides. The nested model value is a framework selector; its own params are not used by that controller.