squ11z1 commited on
Commit
fcf9e7f
·
verified ·
1 Parent(s): 09549cc

Upload generate_visualization.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. generate_visualization.py +54 -0
generate_visualization.py ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """
3
+ Generate English visualization for Chronos o1 1.5B results
4
+ """
5
+
6
+ import numpy as np
7
+ import matplotlib.pyplot as plt
8
+
9
+ np.random.seed(42)
10
+
11
+ train_size = 8
12
+ test_size = 4
13
+
14
+ K_train = np.random.rand(train_size, train_size)
15
+ K_train = (K_train + K_train.T) / 2
16
+ np.fill_diagonal(K_train, 1.0)
17
+
18
+ true_labels = [1, 0, 1, 0]
19
+ predictions = [1, 0, 1, 1]
20
+
21
+ fig, axes = plt.subplots(1, 3, figsize=(15, 4))
22
+
23
+ models = ['Classical\n(Baseline)', 'Quantum\n(Hybrid)']
24
+ accuracies = [1.0, 0.75]
25
+ colors = ['blue', 'red']
26
+
27
+ axes[0].bar(models, accuracies, color=colors, alpha=0.7)
28
+ axes[0].set_ylabel('Accuracy')
29
+ axes[0].set_ylim([0, 1])
30
+ axes[0].set_title('Model Comparison')
31
+ axes[0].grid(True, alpha=0.3)
32
+
33
+ im = axes[1].imshow(K_train, cmap='hot', aspect='auto')
34
+ axes[1].set_title('Quantum Kernel Matrix')
35
+ axes[1].set_xlabel('Sample j')
36
+ axes[1].set_ylabel('Sample i')
37
+ plt.colorbar(im, ax=axes[1])
38
+
39
+ x_pos = np.arange(len(true_labels))
40
+ axes[2].scatter(x_pos, true_labels, c='blue', s=200, alpha=0.5,
41
+ marker='o', label='True')
42
+ axes[2].scatter(x_pos, predictions, c='red', s=100,
43
+ marker='x', label='Predicted')
44
+ axes[2].set_title('Predictions (Quantum Hybrid)')
45
+ axes[2].set_xlabel('Test Sample')
46
+ axes[2].set_ylabel('Class')
47
+ axes[2].set_yticks([0, 1])
48
+ axes[2].set_yticklabels(['Negative', 'Positive'])
49
+ axes[2].legend()
50
+ axes[2].grid(True, alpha=0.3)
51
+
52
+ plt.tight_layout()
53
+ plt.savefig('chronos_o1_results.png', dpi=150, bbox_inches='tight')
54
+ print("Visualization saved: chronos_o1_results.png")