MlModel
Namespace: MLLIB · Library: Machine Learning Library
#include <ml/ml_model.h>
class MLLIB::MlModel
Backend-agnostic inference interface: load, predict, save plus model/task descriptors.
Inheritance
Public Methods
~MlModel()
Virtual destructor.
predict(input)
Run inference on the given input tensor.
Parameters:
- input : const MlTensor & The input data.
Returns:
- MlTensor — The prediction result.
save(path)
Serialise the model to disk.
Parameters:
- path : const QString & File path.
Returns:
- bool — True if successful.
load(path)
Load a model from disk.
Parameters:
- path : const QString & File path.
Returns:
- bool — True if successful.
modelType()
Returns:
- QString — Human-readable model type name.
taskType()
Returns:
- MlTaskType — The task type this model is configured for.
Authors of this file
- Christoph Dinh <christoph.dinh@mne-cpp.org>