INT104-lab9 测试新数据集

 1 import numpy as np
 2 from sklearn.manifold import TSNE
 3 import matplotlib.pyplot as plt
 4 from mpl_toolkits.mplot3d import Axes3D
 5 
 6 
 7 def read(path: str) -> list:
 8     with open(path, "r") as f:
 9         text = f.readlines()
10         D = []
11         for row in text:
12             row = str.replace(row, "14:", ",")
13             row = str.replace(row, "13:", ",")
14             row = str.replace(row, "12:", ",")
15             row = str.replace(row, "11:", ",")
16             row = str.replace(row, "10:", ",")
17             row = str.replace(row, "9:", ",")
18             row = str.replace(row, "8:", ",")
19             row = str.replace(row, "7:", ",")
20             row = str.replace(row, "6:", ",")
21             row = str.replace(row, "5:", ",")
22             row = str.replace(row, "4:", ",")
23             row = str.replace(row, "3:", ",")
24             row = str.replace(row, "2:", ",")
25             row = str.replace(row, "1:", ",")
26             row = str.replace(row, ",-1 1,", ",")
27             row = str.replace(row, ",1 1,", ",")
28             substr = str.split(str.split(row, "
")[0], ",")
29             X = []
30             for a in substr:
31                 X.append(float(a))
32             D.append(X)
33     return D
34 
35 
36 def init(D: list) -> tuple:
37     n, m = len(D), len(D[0])
38     X, Y = [], []
39     for i in range(n):
40         x = []
41         if len(D[i]) != m:
42             n -= 1
43             continue
44         for j in range(1, m, 1):
45             x.append(D[i][j])
46         X.append(x)
47         Y.append(D[i][0])
48     # print(n, m)
49     return X, Y, n, m - 1
50 
51 
52 if __name__ == '__main__':
53 
54     D = read("binary_data.txt")
55     X, Y, n, m = init(D)
56 
57     x = np.array(X)
58 
59     tsne = TSNE(n_components=3)
60 
61     tsne.fit_transform(x)
62 
63     one_x, one_y, one_z, zero_x, zero_y, zero_z = [], [], [], [], [], []
64     for i in range(n):
65         _x, _y, _z = tsne.embedding_[i][0], tsne.embedding_[i][1], tsne.embedding_[i][2]
66         if Y[i] > 0:
67             zero_x.append(_x)
68             zero_y.append(_y)
69             zero_z.append(_z)
70         else:
71             one_x.append(_x)
72             one_y.append(_y)
73             one_z.append(_z)
74     '''
75     ax = plt.axes(projection='3d')
76     ax.scatter3D(one_x, one_y, one_z)
77     ax.scatter3D(zero_x, zero_y, zero_z)
78     '''
79     plt.subplot(311)
80     plt.scatter(one_x, one_y)
81     plt.scatter(zero_x, zero_y)
82     plt.subplot(312)
83     plt.scatter(one_x, one_z)
84     plt.scatter(zero_x, zero_z)
85     plt.subplot(313)
86     plt.scatter(one_y, one_z)
87     plt.scatter(zero_y, zero_z)
88     plt.show()