Jayant-Kernel commited on
fix: save charts to /tmp, upload to 1.5b model repo
Browse files- generate_charts.py +5 -5
generate_charts.py
CHANGED
|
@@ -58,7 +58,7 @@ axes[3].set_ylim(0, 100)
|
|
| 58 |
|
| 59 |
plt.suptitle('DECEIT: Base Model vs Trained Model\n(Qwen 2.5 1.5B, 30 episodes each)', fontsize=13)
|
| 60 |
plt.tight_layout()
|
| 61 |
-
plt.savefig('comparison_chart.png', dpi=150, bbox_inches='tight')
|
| 62 |
plt.close()
|
| 63 |
print("Saved comparison_chart.png")
|
| 64 |
|
|
@@ -78,7 +78,7 @@ ax.annotate('27% reduction\nin sycophancy', xy=(1, 26.7), xytext=(0.5, 45),
|
|
| 78 |
arrowprops=dict(arrowstyle='->', color='black'),
|
| 79 |
fontsize=12, fontweight='bold', color='green')
|
| 80 |
plt.tight_layout()
|
| 81 |
-
plt.savefig('sycophancy_chart.png', dpi=150, bbox_inches='tight')
|
| 82 |
plt.close()
|
| 83 |
print("Saved sycophancy_chart.png")
|
| 84 |
|
|
@@ -97,16 +97,16 @@ ax.annotate('267% increase\nin honest abstention', xy=(1, 36.7), xytext=(0.3, 50
|
|
| 97 |
arrowprops=dict(arrowstyle='->', color='black'),
|
| 98 |
fontsize=12, fontweight='bold', color='green')
|
| 99 |
plt.tight_layout()
|
| 100 |
-
plt.savefig('honesty_chart.png', dpi=150, bbox_inches='tight')
|
| 101 |
plt.close()
|
| 102 |
print("Saved honesty_chart.png")
|
| 103 |
|
| 104 |
# Upload all charts to HF Hub
|
| 105 |
-
for filename in ['comparison_chart.png', 'sycophancy_chart.png', 'honesty_chart.png']:
|
| 106 |
upload_file(
|
| 107 |
path_or_fileobj=filename,
|
| 108 |
path_in_repo=filename,
|
| 109 |
-
repo_id="Ajsaxena/deceit-qwen-
|
| 110 |
repo_type="model"
|
| 111 |
)
|
| 112 |
print(f"Uploaded {filename} to HF Hub")
|
|
|
|
| 58 |
|
| 59 |
plt.suptitle('DECEIT: Base Model vs Trained Model\n(Qwen 2.5 1.5B, 30 episodes each)', fontsize=13)
|
| 60 |
plt.tight_layout()
|
| 61 |
+
plt.savefig('/tmp/comparison_chart.png', dpi=150, bbox_inches='tight')
|
| 62 |
plt.close()
|
| 63 |
print("Saved comparison_chart.png")
|
| 64 |
|
|
|
|
| 78 |
arrowprops=dict(arrowstyle='->', color='black'),
|
| 79 |
fontsize=12, fontweight='bold', color='green')
|
| 80 |
plt.tight_layout()
|
| 81 |
+
plt.savefig('/tmp/sycophancy_chart.png', dpi=150, bbox_inches='tight')
|
| 82 |
plt.close()
|
| 83 |
print("Saved sycophancy_chart.png")
|
| 84 |
|
|
|
|
| 97 |
arrowprops=dict(arrowstyle='->', color='black'),
|
| 98 |
fontsize=12, fontweight='bold', color='green')
|
| 99 |
plt.tight_layout()
|
| 100 |
+
plt.savefig('/tmp/honesty_chart.png', dpi=150, bbox_inches='tight')
|
| 101 |
plt.close()
|
| 102 |
print("Saved honesty_chart.png")
|
| 103 |
|
| 104 |
# Upload all charts to HF Hub
|
| 105 |
+
for filename in ['/tmp/comparison_chart.png', '/tmp/sycophancy_chart.png', '/tmp/honesty_chart.png']:
|
| 106 |
upload_file(
|
| 107 |
path_or_fileobj=filename,
|
| 108 |
path_in_repo=filename,
|
| 109 |
+
repo_id="Ajsaxena/deceit-qwen-1.5b-full",
|
| 110 |
repo_type="model"
|
| 111 |
)
|
| 112 |
print(f"Uploaded {filename} to HF Hub")
|