Spaces:
Sleeping
Sleeping
File size: 1,661 Bytes
bd39f54 |
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 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from coding.llh.static.config import Config
# Draw scatter
def draw_scatter_2D(x_data, y_data, centers, title):
num_clusters = np.unique(y_data)
plt.figure(figsize=(10, 8))
for i in range(len(num_clusters)):
plt.scatter(x_data[y_data == i][:, 0], x_data[y_data == i][:, 1], s=1)
for i in range(len(num_clusters)):
plt.scatter(centers[i, 0], centers[i, 1], marker="*", s=50, c="black")
plt.title(title)
plt.savefig("./diagram/{}.png".format(title), dpi=300)
plt.show()
def draw_scatter_2D_1(x_data, title):
plt.figure(figsize=(10, 8))
plt.scatter(x_data[:, 0], x_data[:, 1], s=1)
plt.title(title)
plt.savefig("./diagram/{}.png".format(title), dpi=300)
plt.show()
def draw_scatter_3D(x_data, y_data, centers, title):
num_clusters = np.unique(y_data)
fig = plt.figure(figsize=(10, 8))
ax = Axes3D(fig)
fig.add_axes(ax)
for i in range(len(num_clusters)):
ax.scatter(x_data[y_data == i][:, 0], x_data[y_data == i][:, 1], x_data[y_data == i][:, 2], s=1)
for i in range(len(num_clusters)):
ax.scatter(centers[i, 0], centers[i, 1], centers[i, 2], marker="*", s=50, c="black")
plt.title(title)
plt.savefig("./diagram/{}.png".format(title), dpi=300)
plt.show()
def draw_scatter_3D_1(x_data, title):
fig = plt.figure(figsize=(10, 8))
ax = Axes3D(fig)
fig.add_axes(ax)
ax.scatter(x_data[:, 0], x_data[:, 1], x_data[:, 2], s=1)
plt.title(title)
plt.savefig("./diagram/{}.png".format(title), dpi=300)
plt.show() |