Cont-Steal

This is a Pytorch implementation of Cont-Steal.

Conduct normal attack

   python3 normal_attack.py --model_type --steal --target_dataset

   python3 contrastive_steal.py --model_type --target_dataset

Name		Name	Last commit message	Last commit date
Latest commit History 19 Commits
.DS_Store		.DS_Store
Linear.py		Linear.py
Loss.py		Loss.py
README.md		README.md
contrastive_steal.py		contrastive_steal.py
dataloader.py		dataloader.py
normal_steal.py		normal_steal.py
surrogate_model.py		surrogate_model.py
target_model.py		target_model.py
test_last.py		test_last.py
test_target.py		test_target.py
train_linear.py		train_linear.py
train_onehot.py		train_onehot.py
train_posteriors.py		train_posteriors.py
train_representation.py		train_representation.py
train_surrogate_linear.py		train_surrogate_linear.py
utils.py		utils.py