8f43927575
examples/ is based on KasperskyLab/knp at 55470ed6 with local changes to the mnist-learn altai/blifat models, cuda-backend CMake and model_loader. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
47 lines
1.6 KiB
C++
47 lines
1.6 KiB
C++
/**
|
|
* @file dataset.h
|
|
* @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.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <knp/framework/data_processing/classification/image.h>
|
|
|
|
#include "model_desc.h"
|
|
|
|
|
|
/**
|
|
* @brief Short name for dataset class.
|
|
*/
|
|
using Dataset = knp::framework::data_processing::classification::images::Dataset;
|
|
|
|
/**
|
|
* @brief Process MNIST dataset for neural network training and inference.
|
|
*
|
|
* @details This function reads raw MNIST dataset files (images and labels), loads them into memory, converts the data
|
|
* into spike representations suitable for neural networks, and splits the dataset into training and inference portions
|
|
* according to the model configuration.
|
|
*
|
|
* @param model_desc model description.
|
|
*
|
|
* @return Processed dataset object ready for training and inference operations.
|
|
*
|
|
* @throws std::runtime_error if required dataset files are not found or cannot be read.
|
|
*/
|
|
Dataset process_dataset(ModelDescription const& model_desc);
|