PyGIP/example.py at main · luoluomei/PyGIP · 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
from datasets import Cora

dataset = Cora()

# >>>>>>>>>> test MEA

# mea = ModelExtractionAttack0(dataset, 0.25)
# mea.attack()


# >>>>>>>>>> test ImperceptibleWM
# from models.defense.OwnerWatermarkingDefense import OwnerWatermarkingDefense
# from utils.dglTopyg import dgl_to_pyg_data
#
# pyg_data = dgl_to_pyg_data(dataset.graph)
#
# defense = OwnerWatermarkingDefense(dataset)
# metrics = defense.defend(pyg_data)
#
# print(metrics)


# >>>>>>>>>> test ImperceptibleWM2
# from models.defense.ImperceptibleOwnerUniqueWatermark import WatermarkByBilevelOptimization
#
# defense = WatermarkByBilevelOptimization(dataset)
# defense.defend()


# >>>>>>>>>> test SurviveWM2
from models.defense.SurviveWM2 import OptimizedWatermarkDefense
from datasets import ENZYMES

dataset = ENZYMES()
defense = OptimizedWatermarkDefense(dataset, 0.25)
defense.defend()