Skip to content

Add InferenceModel for safe production inference #387

@mkumar73

Description

@mkumar73

Add an inference-only model interface for loading fitted DeepTab artifacts and running safe predictions in local, batch, or production environments.

InferenceModel should not replace model.predict(). The normal estimator API remains the main workflow for research and local experimentation. InferenceModel is for deployment use cases where the object should only support loading, validation, runtime inspection, and prediction.

Proposed API

from deeptab import InferenceModel

model = InferenceModel.load(
    "model.deeptab",
    device="cpu",
)

model.validate_input(X_new)
preds = model.predict(X_new)

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type
No fields configured for issues without a type.

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions