Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,615 Bytes
246c106 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
import seaborn as sns
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
# Sample data based on the provided image structure
tasks = ["Passive Dynamics", "Full Dynamics", "Forward Dynamics"]
bar_labels = ["Passive Dynamics", "Full Dynamics", "Forward Dynamics"]
values = np.array([
[6.29],
[5.21],
[5.02],
])
values = np.exp(values)
# Bar colors matching the provided image
bar_colors = ['#a6cee3', '#ffffff', '#a6cee3', '#cab2d6', '#b3b3cc', '#33a02c']
# Plotting the data
fig, ax = plt.subplots(figsize=(5, 3))
# Set bar width and x positions for each group
bar_width = 0.4
x = np.arange(len(tasks))
# Plot each group's bars with the specified colors
for i in range(values.shape[1]):
bars = ax.bar(x + i * bar_width, values[:, i], width=bar_width, color=bar_colors[i], edgecolor='black')
bars[-1].set_color('#cab2d6')
bars[-1].set_edgecolor('black')
for container in ax.containers:
ax.bar_label(container, label_type="edge", fontsize="x-large", fmt="%.1f")
# Set titles, labels, and ticks
# ax.set_title("Zero-Shot Performance Comparison Across Tasks")
ax.set_xlabel("Model", fontsize=14)
ax.set_ylabel("Perplexity", fontsize=14)
ax.set_xticks(x )
ax.set_xticklabels(tasks, fontsize=12)
ax.set_ylim(values.min() - 10, values.max() + 50)
ax.tick_params(axis='x', rotation=15)
# Adding the legend outside the plot area
# ax.legend(loc='upper center', bbox_to_anchor=(0.5, 1.15), ncol=3)
# Display the plot
plt.tight_layout()
# plt.show()
plt.savefig("output/dynamics_ablation.png", dpi=300)
|