Skip to main content
. 2021 Jul 13;21(14):4788. doi: 10.3390/s21144788
Algorithm 1 Network security situation awareness algorithm based on LSTM-DT model.
Input: D, m,Sstate, seed, steps, η, attribute A
Output: Prediction value, loss value and sort results corresponding to test set
1. get Dtr, Dte from D by m
2. St = SSAE (Dtr)
3. Dtr’ = zscore (Stm)
4. create LSTMcell by Sstate
5. connect LSTMnet by LSTMcell
6. initialize LSTMnet by seed
7. for each step in 1:steps
8. P = LSTMforward(Dtr’)
9. Loss=i=1mpiyi2/m2
10. update LSTMnet by Loss and η
11. get LSTMnet*
12. for each j in 0:(n − m − 1)
Pf+i = LSTMnet*(P)
append PO with Pf+i[−1]
13. Pte = de_zscore (PO)
14. update Dte by Pte
15. get Dte
16. Tree = Create root node n
17. for all attribute ADte’ do
Use compute information-theoretic criteria
get abest
18. end for
19. Tree = Create node that tests abest in the root
20. get Dv from Dte’ based on abest
21. for all Dv do
Treev = C4.5(Dv)
Attach Treev to the corresponding branch of Tree
22. end for
23. return sort results(attack type)