-
Python 3.7
-
Pytorch 1.7
The pytorch version of sasa basically replicated the results of the tensorflow version of sasa on the Boiler dataset.
You can run it with the following command.
python main.py
AUC scores on the Boiler data set.
1->2 | 1->3 | 2>1 | 2->3 | 3->1 | 3->2 | mean |
---|---|---|---|---|---|---|
0.7260 | 0.9599 | 0.8575 | 0.9451 | 0.9175 | 0.6835 | 0.8483 |