AnoGraph/code/demo.sh at c9cee155ca8471558fb821f7048cf333ecb61f5b · Stream-AD/AnoGraph · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
make clean
make

if [ $1 == "DARPA" ]; then

  echo "Preprocessing Graph Labels according to time_window and edge_threshold"
  python3 process_data.py DARPA 30 50

  echo "Running AnoGraph"
  ./main anograph DARPA 30 50 2 32

  echo "Running AnoGraph-K"
  ./main anograph_k DARPA 30 50 2 32 5

  echo "Running AnoEdge-G"
  # Algorithm => anoedge_g
  # Dataset => DARPA
  # Rows => 2
  # Buckets => 32
  # Decay factor => 0.9
  ./main anoedge_g DARPA 2 32 0.9

  echo "Running AnoEdge-L"
  # Algorithm => anoedge_g
  # Dataset => DARPA
  # Rows => 2
  # Buckets => 32
  # Decay factor => 0.9
  ./main anoedge_l DARPA 2 32 0.9

  echo "Installing python dependencies"
  pip3 install -r requirements.txt

  echo "Running python metrics"
  python3 metrics.py --dataset DARPA --time_window 30 --edge_threshold 50

fi

if [ $1 == "ISCX" ]; then

  echo "Preprocessing Graph Labels according to time_window and edge_threshold"
  python3 process_data.py ISCX 60 100

  echo "Running AnoGraph"
  ./main anograph ISCX 60 100 2 32

  echo "Running AnoGraph-K"
  ./main anograph_k ISCX 60 100 2 32 5

  echo "Running AnoEdge-G"
  ./main anoedge_g ISCX 2 32 0.9

  echo "Running AnoEdge-L"
  ./main anoedge_l ISCX 2 32 0.9

  python3 metrics.py --dataset ISCX --time_window 60 --edge_threshold 100

fi

if [ $1 == "IDS2018" ]; then

  echo "Preprocessing Graph Labels according to time_window and edge_threshold"
  python3 process_data.py IDS2018 60 100

  echo "Running AnoGraph"
  ./main anograph IDS2018 60 100 2 32

  echo "Running AnoGraph-K"
  ./main anograph_k IDS2018 60 100 2 32 5

  echo "Running AnoEdge-G"
  ./main anoedge_g IDS2018 2 32 0.9

  echo "Running AnoEdge-L"
  ./main anoedge_l IDS2018 2 32 0.9

  python3 metrics.py --dataset IDS2018 --time_window 60 --edge_threshold 100

fi

if [ $1 == "DDOS2019" ]; then

  echo "Preprocessing Graph Labels according to time_window and edge_threshold"
  python3 process_data.py DDOS2019 60 100

  echo "Running AnoGraph"
  ./main anograph DDOS2019 60 100 2 32

  echo "Running AnoGraph-K"
  ./main anograph_k DDOS2019 60 100 2 32 5

  echo "Running AnoEdge-G"
  ./main anoedge_g DDOS2019 2 32 0.9

  echo "Running AnoEdge-L"
  ./main anoedge_l DDOS2019 2 32 0.9

  python3 metrics.py --dataset DDOS2019 --time_window 60 --edge_threshold 100

fi