- Проект собирается с помощью средств CMake
- Для проекта необходима установка библиотеки LibTorch, инструкция по установке: https://pytorch.org/cppdocs/installing.html
и CUDA Toolkit, https://developer.nvidia.com/cuda-downloads - Для корректной работы необходимо передать в командную строку режим работы программы ("training", "testing") и корректный путь к нормализованному датасету в формате .csv.
При режиме работы "training" после окончания обучения в директории создастся файл "policyNet.pt", содержащий веса для обученной нейронной сети.
При режиме работы "testing" после окончания тестирования в директории создастся файл "testResults.txt", содержащий метрики результатов тестирования.
-
Notifications
You must be signed in to change notification settings - Fork 0
do0f/IDS-for-IoT-using-Reinforcement-Learning
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published