ml_trainer.h

Go to the documentation of this file.

//=============================================================================================================
 
#ifndef ML_TRAINER_H
#define ML_TRAINER_H
 
#ifndef WASMBUILD // QProcess (used by PythonRunner) is not available in Qt WASM
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "ml_global.h"
 
#include <utils/python_runner.h>
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QString>
#include <QStringList>
 
//=============================================================================================================
// DEFINE NAMESPACE MLLIB
//=============================================================================================================
 
namespace MLLIB
{
 
//=============================================================================================================
class MLSHARED_EXPORT MLTrainer
{
public:
    //=========================================================================================================
    MLTrainer();
 
    //=========================================================================================================
    explicit MLTrainer(const UTILSLIB::PythonRunnerConfig& config);
 
    //=========================================================================================================
    UTILSLIB::PythonRunner& runner();
 
    //=========================================================================================================
    UTILSLIB::PythonRunnerResult run(const QString& scriptPath,
                                     const QStringList& args = {});
 
    //=========================================================================================================
    QStringList checkPrerequisites(const QStringList& packages) const;
 
private:
    UTILSLIB::PythonRunner m_runner;    
};
 
} // namespace MLLIB
 
#endif // WASMBUILD
 
#endif // ML_TRAINER_H