Skip to content

Commit f61351a

Browse files
committed
fix: labelflipping attack
1 parent ee35561 commit f61351a

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

nebula/addons/attacks/poisoning/labelflipping.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
def labelFlipping(
1717
dataset,
1818
indices,
19-
poisoned_persent=0,
19+
poisoned_percent=0,
2020
targeted=False,
2121
target_label=4,
2222
target_changed_label=7,
@@ -56,9 +56,9 @@ def labelFlipping(
5656
# classes = new_dataset.classes
5757
# class_to_idx = new_dataset.class_to_idx
5858
# class_list = [class_to_idx[i] for i in classes]
59-
class_list = set(targets.tolist())
59+
class_list = list(set(targets.tolist()))
6060
if not targeted:
61-
num_flipped = int(poisoned_persent * num_indices)
61+
num_flipped = int(poisoned_percent * num_indices)
6262
if num_indices == 0:
6363
return new_dataset
6464
if num_flipped > num_indices:

nebula/frontend/templates/deployment.html

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2507,10 +2507,10 @@ <h5 class="step-title">Schema of deployment</h5>
25072507
document.getElementById("poisoned-node-title").style.display = "block";
25082508
document.getElementById("poisoned-node-percent-container").style.display = "block";
25092509
document.getElementById("poisoned-sample-percent").value = 0;
2510-
document.getElementById("poisoned-sample-title").style.display = "none";
2511-
document.getElementById("poisoned-sample-percent-container").style.display = "none";
2512-
document.getElementById("poisoned-noise-title").style.display = "block";
2513-
document.getElementById("poisoned-noise-percent-container").style.display = "block";
2510+
document.getElementById("poisoned-sample-title").style.display = "block";
2511+
document.getElementById("poisoned-sample-percent-container").style.display = "block";
2512+
document.getElementById("poisoned-noise-title").style.display = "none";
2513+
document.getElementById("poisoned-noise-percent-container").style.display = "none";
25142514
}
25152515
else if(this.value == "GLLNeuronInversionAttack" || this.value == "NoiseInjectionAttack" || this.value == "SwappingWeightsAttack" || this.value == "DelayerAttack"){
25162516
document.getElementById("poisoned-node-title").style.display = "block";

0 commit comments

Comments
 (0)