np.choose(y, [1, 2, 0]).astype(np.float)程式問題 - Cupoy

2020/05/12 02:09 下午

機器學習共學討論版

吳宗翰

觀看數：7

回答數：2

收藏數：0

請問下面程式碼中紅色這一行的作用是什麼?

去掉對結果似乎沒有影響 ( 抱歉自己看不太出來 )

# Plot the ground truth

fig = plt.figure(fignum, figsize=(12, 6))

ax = Axes3D(fig, rect=[0, 0, .95, 1], elev=20, azim=100)

for name, label in [('cls0', 0),

                    ('cls1', 1),

                    ('cls2', 2)]:

    ax.text3D(X[y == label, 3].mean(),

              X[y == label, 0].mean(),

              X[y == label, 2].mean() + 2, name,

              horizontalalignment='center',

              bbox=dict(alpha=.2, edgecolor='w', facecolor='w'))

# Reorder the labels to have colors matching the cluster results

y = np.choose(y, [1, 2, 0]).astype(np.float)

ax.scatter(X[:, 3], X[:, 0], X[:, 2], c=y, edgecolor='k',s = 80)

ax.w_xaxis.set_ticklabels([])

ax.w_yaxis.set_ticklabels([])

ax.w_zaxis.set_ticklabels([])

ax.set_title('Ground Truth')

ax.dist = 12

#fig.show()