NLPV commited on
Commit
0a49c8e
Β·
verified Β·
1 Parent(s): 09f5386

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -7
app.py CHANGED
@@ -139,14 +139,16 @@ def run_unlearning(index_to_unlearn, learning_rate):
139
 
140
  # Prediction before unlearning
141
  probs_before, pred_before = predict(original_model, image_tensor)
142
- conf_before = probs_before[label_idx].item()
 
143
 
144
  # Perform the unlearning process on the duplicated model
145
  unlearn(unlearned_model, image_tensor, label_idx, learning_rate)
146
 
147
  # Prediction after unlearning
148
  probs_after, pred_after = predict(unlearned_model, image_tensor)
149
- conf_after = probs_after[label_idx].item()
 
150
 
151
  print("Post-unlearning probabilities:", probs_after)
152
 
@@ -157,13 +159,17 @@ def run_unlearning(index_to_unlearn, learning_rate):
157
  result = f"""
158
  πŸ“ Index Unlearned: {index_to_unlearn}
159
  πŸ—‚οΈ Actual Label: {label_name} (Index: {label_idx})
 
160
  πŸ”Ž BEFORE Unlearning:
161
- - Prediction: {cifar10_classes[pred_before]}
162
- - Confidence: {conf_before:.10f}
 
163
  🧽 AFTER Unlearning:
164
- - Prediction: {cifar10_classes[pred_after]}
165
- - Confidence: {conf_after:.10f}
166
- πŸ“‰ Confidence Drop: {conf_before - conf_after:.6f}
 
 
167
  πŸ§ͺ Test Set Performance:
168
  - Original Model: {orig_acc:.2f}% accuracy, Loss: {orig_loss:.4f}
169
  - Unlearned Model: {unlearn_acc:.2f}% accuracy, Loss: {unlearn_loss:.4f}
 
139
 
140
  # Prediction before unlearning
141
  probs_before, pred_before = predict(original_model, image_tensor)
142
+ conf_actual_before = probs_before[label_idx].item()
143
+ conf_pred_before = probs_before[pred_before].item()
144
 
145
  # Perform the unlearning process on the duplicated model
146
  unlearn(unlearned_model, image_tensor, label_idx, learning_rate)
147
 
148
  # Prediction after unlearning
149
  probs_after, pred_after = predict(unlearned_model, image_tensor)
150
+ conf_actual_after = probs_after[label_idx].item()
151
+ conf_pred_after = probs_after[pred_after].item()
152
 
153
  print("Post-unlearning probabilities:", probs_after)
154
 
 
159
  result = f"""
160
  πŸ“ Index Unlearned: {index_to_unlearn}
161
  πŸ—‚οΈ Actual Label: {label_name} (Index: {label_idx})
162
+
163
  πŸ”Ž BEFORE Unlearning:
164
+ - Predicted Class: {cifar10_classes[pred_before]} with confidence: {conf_pred_before:.10f}
165
+ - Actual Class: {label_name} with confidence: {conf_actual_before:.10f}
166
+
167
  🧽 AFTER Unlearning:
168
+ - Predicted Class: {cifar10_classes[pred_after]} with confidence: {conf_pred_after:.10f}
169
+ - Actual Class: {label_name} with confidence: {conf_actual_after:.10f}
170
+
171
+ πŸ“‰ Confidence Drop (Actual Class): {conf_actual_before - conf_actual_after:.6f}
172
+
173
  πŸ§ͺ Test Set Performance:
174
  - Original Model: {orig_acc:.2f}% accuracy, Loss: {orig_loss:.4f}
175
  - Unlearned Model: {unlearn_acc:.2f}% accuracy, Loss: {unlearn_loss:.4f}