Added STDP AltAI example, and refactored example accordingly.

examples/mnist-learn/CMakeLists.txt

@@ -17,15 +17,27 @@ limitations under the License.
 cmake_minimum_required(VERSION 3.22)
 project(mnist_learn_example)
 
-set(CMAKE_CXX_STANDARD 20)
+set(CMAKE_CXX_STANDARD 17)
 set(CMAKE_VERBOSE_MAKEFILE ON)
 
-find_package(Boost COMPONENTS program_options REQUIRED )
-
-# Need to check if the example build is inside the Kaspersky Neuromorphic Platform tree.
+# Need to check if the example build inside KNP tree.
 if (NOT TARGET KNP::BaseFramework::Core)
     find_package(knp-base-framework REQUIRED)
 endif()
 
-add_executable(mnist_learn_example main.cpp construct_network.cpp train.cpp inference.cpp time_string.cpp)
-target_link_libraries(mnist_learn_example PRIVATE KNP::BaseFramework::Core ${Boost_LIBRARIES})
+find_package(Boost ${KNP_BOOST_MIN_VERSION} COMPONENTS program_options REQUIRED)
+
+add_executable(${PROJECT_NAME}
+    main.cpp
+    model_desc.cpp
+    parse_arguments.cpp
+    dataset.cpp
+    evaluate_results.cpp
+    models/blifat/construct_network.cpp
+    models/blifat/finalize_network.cpp
+    models/blifat/spike_generators.cpp
+    models/altai/construct_network.cpp
+    models/altai/finalize_network.cpp
+    models/altai/spike_generators.cpp)
+target_include_directories(${PROJECT_NAME} PRIVATE ${PROJECT_SOURCE_DIR})
+target_link_libraries(${PROJECT_NAME} PRIVATE KNP::BaseFramework::Core Boost::program_options)

examples/mnist-learn/construct_network.h → examples/mnist-learn/annotated_network.h

@@ -1,10 +1,10 @@
 /**
- * @file construct_network.h
- * @brief Functions for network construction.
- * @kaspersky_support A. Vartenkov
- * @date 03.12.2024
+ * @file annotated_network.h
+ * @brief Network with annotations.
+ * @kaspersky_support D. Postnikov
+ * @date 03.02.2026
  * @license Apache 2.0
- * @copyright © 2024-2025 AO Kaspersky Lab
+ * @copyright © 2026 AO Kaspersky Lab
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -23,39 +23,55 @@
 
 #include <knp/framework/network.h>
 
-#include <filesystem>
 #include <map>
 #include <set>
 #include <string>
 #include <utility>
 #include <vector>
 
 
+/// A network with annotations.
 struct AnnotatedNetwork
 {
+    /// Network.
     knp::framework::Network network_;
+
+    /// Annotation struct.
     struct Annotation
     {
+        /// Output uids.
         // cppcheck-suppress unusedStructMember
         std::vector<knp::core::UID> output_uids_;
+
+        /// Projections from rasterized channeled population.
         // cppcheck-suppress unusedStructMember
         std::vector<knp::core::UID> projections_from_raster_;
+
+        /// Projection from classes channeled populations. Aka label populations.
         // cppcheck-suppress unusedStructMember
         std::vector<knp::core::UID> projections_from_classes_;
+
+        /// Populations that should be kept in inference.
         // cppcheck-suppress unusedStructMember
         std::set<knp::core::UID> inference_population_uids_;
+
+        /// Projections that should be kept in inference.
         // cppcheck-suppress unusedStructMember
         std::set<knp::core::UID> inference_internal_projection_;
 
-        // For each compound network: a vector of senders and a vector of receivers.
+        /// For each compound network: a vector of senders and a vector of receivers.
         // cppcheck-suppress unusedStructMember
         std::vector<std::pair<std::vector<knp::core::UID>, std::vector<knp::core::UID>>> wta_data_;
+
+        /// WTA borders, for example 2,4,6.
+        // cppcheck-suppress unusedStructMember
+        std::vector<size_t> wta_borders_;
+
+        /// Map of population names.
         // cppcheck-suppress unusedStructMember
         std::map<knp::core::UID, std::string> population_names_;
     }
+    /// Annotation to network.
     // cppcheck-suppress unusedStructMember
     data_;
 };
-
-
-AnnotatedNetwork create_example_network(int num_compound_networks);

examples/mnist-learn/dataset.cpp

@@ -0,0 +1,50 @@
+/**
+ * @file dataset.cpp
+ * @brief Process dataset.
+ * @kaspersky_support D. Postnikov
+ * @date 03.02.2026
+ * @license Apache 2.0
+ * @copyright © 2026 AO Kaspersky Lab
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "dataset.h"
+
+#include <fstream>
+
+#include "global_config.h"
+
+
+Dataset process_dataset(ModelDescription const& model_desc)
+{
+    // Create streams for images and labels.
+    std::ifstream images_stream(model_desc.images_file_path_, std::ios::binary);
+    std::ifstream labels_stream(model_desc.labels_file_path_, std::ios::in);
+
+    Dataset dataset;
+    // Process them.
+    dataset.process_labels_and_images(
+        images_stream, labels_stream, model_desc.train_images_amount_ + model_desc.inference_images_amount_,
+        classes_amount, input_size, steps_per_image,
+        dataset.make_incrementing_image_to_spikes_converter(active_steps, state_increment_factor));
+    // Split dataset according to model description.
+    dataset.split(model_desc.train_images_amount_, model_desc.inference_images_amount_);
+
+    // Print out results.
+    std::cout << "Processed dataset, training will last " << dataset.get_steps_amount_for_training()
+              << " steps, inference " << dataset.get_steps_amount_for_inference() << " steps\n"
+              << std::endl;
+
+    return dataset;
+}