Spaces:
Running
Running
File size: 1,816 Bytes
2dd1349 |
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 |
# Call Matplotlib Library to draw graphs (Bar/Plot...)
import matplotlib.pyplot as plt
from llm import *
class Plotting:
def bar_plot(self, data: dict, title: str, xlabel: str, ylabel: str, output_path: str = 'bar_plot.png'):
"""
Create a bar plot.
:param data: Dictionary containing data to plot (keys as labels, values as heights).
:param title: Title of the plot.
:param xlabel: Label for the X-axis.
:param ylabel: Label for the Y-axis.
:param output_path: Path to save the plot image.
"""
labels = list(data.keys())
heights = list(data.values())
plt.figure(figsize=(10, 6))
plt.bar(labels, heights, color='skyblue')
plt.title(title)
plt.xlabel(xlabel)
plt.ylabel(ylabel)
plt.tight_layout()
plt.savefig(output_path)
plt.close()
return output_path
def line_plot(self, data: dict, title: str, xlabel: str, ylabel: str, output_path: str = 'line_plot.png'):
"""
Create a line plot.
:param data: Dictionary containing data to plot (keys as x-values, values as y-values).
:param title: Title of the plot.
:param xlabel: Label for the X-axis.
:param ylabel: Label for the Y-axis.
:param output_path: Path to save the plot image.
"""
x_values = list(data.keys())
y_values = list(data.values())
plt.figure(figsize=(10, 6))
plt.plot(x_values, y_values, marker='o', color='skyblue')
plt.title(title)
plt.xlabel(xlabel)
plt.ylabel(ylabel)
plt.grid(True)
plt.tight_layout()
plt.savefig(output_path)
plt.close()
return output_path
def get_plot(self, data):
instruction = ""
|