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>
52 lines
1.8 KiB
C++
52 lines
1.8 KiB
C++
/**
|
|
* @file wta.cpp
|
|
* @brief Functions for Winner Takes All implementation.
|
|
* @kaspersky_support D. Postnikov
|
|
* @date 03.07.2025
|
|
* @license Apache 2.0
|
|
* @copyright © 2025 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 <knp/framework/monitoring/observer.h>
|
|
#include <knp/framework/projection/wta.h>
|
|
#include <knp/framework/sonata/network_io.h>
|
|
#include <knp/synapse-traits/all_traits.h>
|
|
|
|
namespace knp::framework::projection
|
|
{
|
|
|
|
|
|
std::vector<knp::core::UID> add_wta_handlers(
|
|
knp::framework::ModelExecutor& executor, size_t winners_amount, std::vector<size_t> const& borders,
|
|
std::vector<std::pair<std::vector<knp::core::UID>, std::vector<knp::core::UID>>> const& wta_data)
|
|
{
|
|
std::vector<knp::core::UID> result;
|
|
|
|
// Generating seed for WTA randomness
|
|
std::mt19937 rand_gen(std::random_device{}());
|
|
std::uniform_int_distribution<int> distr(-std::numeric_limits<int>::max(), std::numeric_limits<int>::max());
|
|
|
|
for (const auto& senders_receivers : wta_data)
|
|
{
|
|
knp::core::UID handler_uid;
|
|
executor.add_spike_message_handler(
|
|
knp::framework::modifier::KWtaPerGroup{borders, winners_amount, distr(rand_gen)}, senders_receivers.first,
|
|
senders_receivers.second, handler_uid);
|
|
result.push_back(handler_uid);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
} // namespace knp::framework::projection
|