修改随机森林的训练逻辑

SigerCalculation/RadomForestSAC/RamdomForestCalculate.cpp

@@ -5,7 +5,8 @@
 //发送给流式的消息
 RamdomForestCalculate::RamdomForestCalculate(QObject *parent)
 {
-    SPDLOG_LOGGER_DEBUG(spdlog::get("logger"),"Hello RamdomForestCalculate ");
+    QString a = "算法模块开始运行";
+    SPDLOG_LOGGER_DEBUG(spdlog::get("logger"),"{}",a.toStdString());
 
     qRegisterMetaType<Dc_SacData>("Dc_SacData");
 
@@ -22,25 +23,33 @@ RamdomForestCalculate::RamdomForestCalculate(QObject *parent)
     QString dbPath = QString("/home/pi/SigerTMS/stream/SAC/%1.db").arg("RamdomForestFeatureValue");
     QList<SACRecord> record = queryRecordsByLabel(dbPath,0);
 
-    std::vector<std::vector<float>> FearValues;
-    std::vector<int> TypeValues;
-    DealRecord(record,FearValues,TypeValues);
-    Classifier->train(FearValues, TypeValues, 0.2f, 100);
+    if(record.size() != 0)
+    {
+        IfHaveModel = true;
+        std::vector<std::vector<float>> FearValues;
+        std::vector<int> TypeValues;
+        DealRecord(record,FearValues,TypeValues);
+        Classifier->train(FearValues, TypeValues, 0.2f, 100);
+    }
+
 
     //连接信号和槽函数
     QObject::connect(m_tcpServerManager, &TcpServerManager::getdc_sacdata,
                      [=](const Dc_SacData &sac_data) {
-        qDebug() << "收到 dc_sacdata 信号";
-        qDebug() << "Source:" << sac_data.source;
-        qDebug() << "SacFile:" << sac_data.sacFile;
-        qDebug() << "Target:" << sac_data.target;
+
+        SPDLOG_LOGGER_DEBUG(spdlog::get("logger"),"收到 dc_sacdata 信号");
+        SPDLOG_LOGGER_DEBUG(spdlog::get("logger"),"Source:{}",sac_data.source.toStdString());
+        SPDLOG_LOGGER_DEBUG(spdlog::get("logger"),"SacFile:{}",sac_data.sacFile.toStdString());
+        SPDLOG_LOGGER_DEBUG(spdlog::get("logger"),"Target:{}",sac_data.target.toStdString());
+
         Save_SACDATA(sac_data);  // 现在可以识别了
     });
 
     //连接TMS信号和槽函数
     QObject::connect(m_tcpServerManager, &TcpServerManager::TMSTrain,
                      [=]() {
-        qDebug() << "收到 TMS 信号";
+        SPDLOG_LOGGER_DEBUG(spdlog::get("logger"),"收到 TMS 信号");
+        IfHaveModel = true;
         Retrain();
         //重新训练
     });
@@ -53,10 +62,10 @@ void RamdomForestCalculate::SendStreamAlarminfo(const StreamInfo &alarmInfo)
 
     // 统一的清理函数
     auto cleanup = [client]() {
-        static bool cleaned = false;  // 静态变量确保只清理一次
+        static bool cleaned = false;  // 静态变量确保只清理一次收到 TMS 信号
         if (!cleaned) {
             cleaned = true;
-            qDebug() << "清理客户端资源";
+            //qDebug() << "清理客户端资源";
             client->disconnectFromServer();
             QTimer::singleShot(0, client, &QObject::deleteLater);  // 延迟删除确保安全
         }
@@ -74,7 +83,7 @@ void RamdomForestCalculate::SendStreamAlarminfo(const StreamInfo &alarmInfo)
         message["endIds"] = alarmInfo.endIds;
 
         client->sendMessage(message);
-        qDebug() << "消息发送完成";
+        SPDLOG_LOGGER_DEBUG(spdlog::get("logger"),"发送给流式告警信息成功");
 
         // 发送完成后立即开始清理
         cleanup();
@@ -173,24 +182,28 @@ int RamdomForestCalculate::Save_SACDATA(const Dc_SacData &sacdata)
     //step2: 预测报警
     //std::string data;
     qDebug()<<"x "<<x;
-    std::vector<float> test_sample(10,0);
-    //用逗号分割
-    QStringList list = x.split(',');
-
-    //转成 float 并存到 vector
-    for (const QString &s : list) {
-        bool ok = false;
-        float f = s.toFloat(&ok);
-        if (ok) {
-            test_sample.push_back(f);
-        } else {
-            std::cerr << "转换失败: " << s.toStdString() << std::endl;
+
+    if(IfHaveModel)
+    {
+        std::vector<float> test_sample(10,0);
+        //用逗号分割
+        QStringList list = x.split(',');
+
+        //转成 float 并存到 vector
+        for (const QString &s : list) {
+            bool ok = false;
+            float f = s.toFloat(&ok);
+            if (ok) {
+                test_sample.push_back(f);
+            } else {
+                std::cerr << "转换失败: " << s.toStdString() << std::endl;
+            }
         }
-    }
-    double pred = Classifier->predict(test_sample);
-    cout << "测试样本预测类别: " << pred << endl;
+        double pred = Classifier->predict(test_sample);
+        cout << "测试样本预测类别: " << pred << endl;
 
-    StreamInfo(pred,sacdata.target,sacdata.startIds,sacdata.endIds);
+        StreamInfo(pred,sacdata.target,sacdata.startIds,sacdata.endIds);
+    }
 
     //step3: 存入数据库
     std::string data = QString(

SigerCalculation/RadomForestSAC/RamdomForestCalculate.h

@@ -86,6 +86,7 @@ private:
     TcpServerManager *m_tcpServerManager;
 
     RandomForestClassifier* Classifier;
+    bool IfHaveModel = false;
 };