ml_model.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ML_MODEL_H
#define ML_MODEL_H
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "ml_global.h"
#include "ml_types.h"
#include "ml_tensor.h"
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QSharedPointer>
#include <QString>
 
//=============================================================================================================
// DEFINE NAMESPACE MLLIB
//=============================================================================================================
 
namespace MLLIB
{
 
//=============================================================================================================
class MLSHARED_EXPORT MlModel
{
public:
    typedef QSharedPointer<MlModel> SPtr; 
 
    //=========================================================================================================
    virtual ~MlModel() = default;
 
    //=========================================================================================================
    virtual MlTensor predict(const MlTensor& input) const = 0;
 
    //=========================================================================================================
    virtual bool save(const QString& path) const = 0;
 
    //=========================================================================================================
    virtual bool load(const QString& path) = 0;
 
    //=========================================================================================================
    virtual QString modelType() const = 0;
 
    //=========================================================================================================
    virtual MlTaskType taskType() const = 0;
};
 
} // namespace MLLIB
 
#endif // ML_MODEL_H