Add plot

README.md

@@ -1,5 +1,8 @@
 This repo contains sparsity report for each of the pruned model in the table below.
 
+The report (csv) shows layer-wise sparsity, sparsity by tile of 128x16, sparsity by col and row global to its layers.
+
+
 Pruning ```meta-llama/Meta-Llama-3.1-8B``` with Wanda
 | Weight Target Sparsity | Perplexity (lower is better) |
 |------------------------|-----------------------------|
@@ -12,7 +15,8 @@ Pruning ```meta-llama/Meta-Llama-3.1-8B``` with Wanda
 | 60                      | 20.2265                     |
 | 70                      | 103.5209                    |
 
-The report (csv) shows layer-wise sparsity, sparsity by tile of 128x16, sparsity by col and row global to its layers.
+![Perplexity over Sparsity](./perplexity_vs_sparsity.png)
+
 
 > For a more granular sparsity report within a given tile, pls continue below
 

plot.py

@@ -0,0 +1,19 @@
+import matplotlib.pyplot as plt
+
+# Data from the table
+sparsity = [0, 10, 20, 30, 40, 50, 60, 70]
+perplexity = [5.8393, 5.8781, 6.0102, 6.3076, 7.0094, 9.0642, 20.2265, 103.5209]
+
+# Final adjustment: plot size to 6x4 inches
+plt.figure(figsize=(6, 4))
+plt.plot(sparsity, perplexity, marker='o', linestyle='-', color='b')
+plt.axhline(y=5.8393, color='g', linestyle='--', label='Perplexity at 0% Sparsity')
+plt.title("Perplexity vs. Weight Target Sparsity", fontsize=14)
+plt.xlabel("Weight Target Sparsity (%)", fontsize=12)
+plt.ylabel("Perplexity (lower is better)", fontsize=12)
+plt.legend(fontsize=10)
+plt.grid(True)
+plt.xticks(fontsize=10)
+plt.yticks(fontsize=10)
+plt.show()
+plt.savefig("perplexity_vs_sparsity.png")