sunsili 发表于 2023-7-14 10:20:58

Kafka +深度学习+ MQTT搭建可扩展的物联网平台【附源码】

Kafka +深度学习+ MQTT搭建可扩展的物联网平台【附源码】




物联网+大数据+机器学习将会是以后的趋势,这里介绍一篇这方面的文章包含源码。混合机器学习基础架构构建了一个场景,利用Apache Kafka作为可扩展的中枢神经系统。 公共云用于极大规模地训练分析模型(例如,通过Google ML Engine在Google Cloud Platform(GCP)上使用TensorFlow和TPU,预测(即模型推断)在本地Kafka基础设施的执行( 例如,利用Kafka Streams或KSQL进行流分析)。
本文重点介绍内部部署。 创建了一个带有KSQL UDF的Github项目,用于传感器分析。 它利用KSQL的新API功能,使用Java轻松构建UDF / UDAF函数,对传入事件进行连续流处理。使用案例:Connected Cars - 使用深度学习的实时流分析从连接设备(本例中的汽车传感器)连续处理数百万个事件:

为此构建了不同的分析模型。 他们在公共云上接受TensorFlow,H2O和Google ML Engine的训练。 模型创建不是此示例的重点。 最终模型已经可以投入生产,可以部署用于实时预测。
模型服务可以通过模型server 完成,也可以本地嵌入到流处理应用程序中。 参阅RPC与流处理的权衡,以获得模型部署和....
演示:使用MQTT,Kafka和KSQL在Edge进行模型推理
Github项目:深度学习+KSQL UDF 用于流式异常检测MQTT物联网传感器数据该项目的重点是通过MQTT将数据提取到Kafka并通过KSQL处理数据:

Confluent MQTT Proxy的一大优势是无需MQTT Broker即可实现物联网方案的简单性。 可以通过MQTT代理将消息直接从MQTT设备转发到Kafka。 这显着降低了工作量和成本。 如果你“只是”想要在Kafka和MQTT设备之间进行通信,这是一个完美的解决方案。如果你想看到另一部分(与Elasticsearch / Grafana等接收器应用程序集成),请查看Github项目“KSQL for streaming IoT data”。 这实现了通过Kafka Connect和Elastic连接器与ElasticSearch和Grafana的集成。KSQL UDF - 源代码开发UDF非常容易。 只需在UDF类中的一个Java方法中实现该函数: 纯文本查看 复制代码@Udf(description = "apply analytic model to sensor input")
这里是所有代码: 纯文本查看 package com.github.megachucky.kafka.streams.machinelearning;

import java.util.Arrays;

import hex.genmodel.GenModel;
import hex.genmodel.easy.EasyPredictModelWrapper;
import hex.genmodel.easy.RowData;
import hex.genmodel.easy.exception.PredictException;
import hex.genmodel.easy.prediction.AutoEncoderModelPrediction;
import io.confluent.ksql.function.udf.Udf;
import io.confluent.ksql.function.udf.UdfDescription;


@UdfDescription(name = "anomaly", description = "anomaly detection using deep learning")
public class Anomaly {
   
   
    // Model built with H2O R API:
      // anomaly_model <- h2o.deeplearning(x = names(train_ecg),training_frame =
      // train_ecg,activation = "Tanh",autoencoder = TRUE,hidden =
      // c(50,20,50),sparse = TRUE,l1 = 1e-4,epochs = 100)

      // Name of the generated H2O model
      private static String modelClassName = "io.confluent.ksql.function.udf.ml"
                                             + ".DeepLearning_model_R_1509973865970_1";
   
@Udf(description = "apply analytic model to sensor input")
public String anomaly(String sensorinput) {
      
      System.out.println("Kai: DL-UDF starting");
         
      GenModel rawModel;
      try {
            rawModel = (hex.genmodel.GenModel) Class.forName(modelClassName).newInstance();
         
      EasyPredictModelWrapper model = new EasyPredictModelWrapper(rawModel);
         
      // Prepare input sensor data to be in correct data format for the autoencoder model (double[]):
      String[] inputStringArray = sensorinput.split("#");
      double[] doubleValues = Arrays.stream(inputStringArray)
                .mapToDouble(Double::parseDouble)
                .toArray();
         
      RowData row = new RowData();
      int j = 0;
      for (String colName : rawModel.getNames()) {
          row.put(colName, doubleValues);
          j++;
      }

      AutoEncoderModelPrediction p = model.predictAutoEncoder(row);
      // System.out.println("original: " + java.util.Arrays.toString(p.original));
      // System.out.println("reconstructedrowData: " + p.reconstructedRowData);
      // System.out.println("reconstructed: " + java.util.Arrays.toString(p.reconstructed));

      double sum = 0;
      for (int i = 0; i < p.original.length; i++) {
          sum += (p.original - p.reconstructed) * (p.original - p.reconstructed);
      }
      // Calculate Mean Square Error => High reconstruction error means anomaly
      double mse = sum / p.original.length;
      System.out.println("MSE: " + mse);

      String mseString = "" + mse;

      return (mseString);
         
      } catch (InstantiationException | IllegalAccessException | ClassNotFoundException e) {
            System.out.println(e.toString());
            
      } catch (PredictException e) {
            System.out.println(e.toString());
      }
         
      return null;      
}
}
如何使用Apache Kafka和MQTT Proxy运行演示?执行演示的所有步骤都在Github项目中描述。你只需安装Confluent Platform,然后按照以下步骤部署UDF,创建MQTT事件并通过KSQL levera处理它们....这里使用Mosquitto生成MQTT消息。 当然,也可以使用任何其他MQTT客户端。 这是开放和标准化协议的巨大好处。
到此结束,文章虽然简短,但是内容确实很丰富,特别项目的源码的阅读,在github上有详细的介绍。github地址:(回复可查看)**** Hidden Message *****
源码下载:(回复可查看)**** Hidden Message *****


页: [1]
查看完整版本: Kafka +深度学习+ MQTT搭建可扩展的物联网平台【附源码】