readme.md

Machine Learning + Kafka Streams Examples

General info in main Readme

Example 3 - Iris Prediction using a Neural Network with DeepLearning4J (DL4J)

Use Case

Iris Species Prediction using a Neural Network. This is a famous example: Prediction of the Iris Species - implemented with many different ML algorithms. Here I use DeepLearning4J (DL4J) to build a neural network using Iris Dataset.

Machine Learning Technology

• DeepLearning4J
• Pretty simple example to demo how to build, save and load neural networks with DL4J. MultiLayerNetwork and INDArray are the key APIs to look at if you want to understand the details.
• The model is created via DeepLearning4J_CSV_Model.java and stored in the resources: DL4J_Iris_Model.zip. No need to re-train, just for reference. Kudos to Adam Gibson who created this example as part of the DL4J project.

Unit Test Kafka_Streams_MachineLearning_DL4J_DeepLearning_Iris_IntegrationTest.java