-
Notifications
You must be signed in to change notification settings - Fork 23
/
driver.py
21 lines (17 loc) · 764 Bytes
/
driver.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import random
import netshare.ray as ray
from netshare import Generator
if __name__ == '__main__':
# Change to False if you would not like to use Ray
ray.config.enabled = False
ray.init(address="auto")
# configuration file
generator = Generator(config="config_example_netflow_nodp.json")
# `work_folder` should not exist o/w an overwrite error will be thrown.
# Please set the `worker_folder` as *absolute path*
# if you are using Ray with multi-machine setup
# since Ray has bugs when dealing with relative paths.
generator.train(work_folder=f'../../results/test-ugr16')
generator.generate(work_folder=f'../../results/test-ugr16')
generator.visualize(work_folder=f'../../results/test-ugr16')
ray.shutdown()