mirror of
https://github.com/clearml/clearml
synced 2025-01-31 17:17:00 +00:00
37 lines
789 B
Python
37 lines
789 B
Python
|
# TRAINS - Example of Matplotlib integration and reporting
|
||
|
|
#
|
||
|
|
import numpy as np
|
||
|
|
import matplotlib.pyplot as plt
|
||
|
|
from trains import Task
|
||
|
|
|
||
|
|
|
||
|
|
task = Task.init(project_name='examples', task_name='Matplotlib example')
|
||
|
|
|
||
|
|
# create plot
|
||
|
|
N = 50
|
||
|
|
x = np.random.rand(N)
|
||
|
|
y = np.random.rand(N)
|
||
|
|
colors = np.random.rand(N)
|
||
|
|
area = (30 * np.random.rand(N))**2 # 0 to 15 point radii
|
||
|
|
plt.scatter(x, y, s=area, c=colors, alpha=0.5)
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
# create another plot - with a name
|
||
|
|
x = np.linspace(0, 10, 30)
|
||
|
|
y = np.sin(x)
|
||
|
|
plt.plot(x, y, 'o', color='black')
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
# create image plot
|
||
|
|
m = np.eye(256, 256, dtype=np.uint8)
|
||
|
|
plt.imshow(m)
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
# create image plot - with a name
|
||
|
|
m = np.eye(256, 256, dtype=np.uint8)
|
||
|
|
plt.imshow(m)
|
||
|
|
plt.title('Image Title')
|
||
|
|
plt.show()
|
||
|
|
|
||
|
|
print('This is a Matplotlib example')
|