ml_trainer.cpp

Go to the documentation of this file.

//=============================================================================================================
 
//=============================================================================================================
// INCLUDES
//=============================================================================================================
 
#include "ml_trainer.h"
 
#ifndef WASMBUILD // QProcess (used by PythonRunner) is not available in Qt WASM
 
//=============================================================================================================
// QT INCLUDES
//=============================================================================================================
 
#include <QDebug>
 
//=============================================================================================================
// USED NAMESPACES
//=============================================================================================================
 
using namespace MLLIB;
using namespace UTILSLIB;
 
//=============================================================================================================
// DEFINE MEMBER METHODS
//=============================================================================================================
 
MLTrainer::MLTrainer()
    : m_runner()
{
}
 
//=============================================================================================================
 
MLTrainer::MLTrainer(const PythonRunnerConfig& config)
    : m_runner(config)
{
}
 
//=============================================================================================================
 
PythonRunner& MLTrainer::runner()
{
    return m_runner;
}
 
//=============================================================================================================
 
PythonRunnerResult MLTrainer::run(const QString& scriptPath,
                                   const QStringList& args)
{
    if (!m_runner.isPythonAvailable()) {
        PythonRunnerResult result;
        result.stdErr = QStringLiteral("Python interpreter not found: ") + m_runner.config().pythonExe;
        qWarning() << "[MLTrainer]" << result.stdErr;
        return result;
    }
 
    qDebug() << "[MLTrainer] Running training script:" << scriptPath;
 
    // If a venv is configured, use runInVenv (creates venv + installs deps automatically)
    if (!m_runner.config().venvDir.isEmpty()) {
        return m_runner.runInVenv(scriptPath, args);
    }
 
    return m_runner.run(scriptPath, args);
}
 
//=============================================================================================================
 
QStringList MLTrainer::checkPrerequisites(const QStringList& packages) const
{
    QStringList missing;
    for (const QString& pkg : packages) {
        if (!m_runner.isPackageAvailable(pkg)) {
            missing << pkg;
        }
    }
    if (!missing.isEmpty()) {
        qWarning() << "[MLTrainer] Missing Python packages:" << missing.join(QStringLiteral(", "));
    }
    return missing;
}
 
#endif // WASMBUILD