tsne.py 文件源码

def main1():
    #target_names = np.array(args.names)
    X, Y = getFeatureAndPath(args.image_dir)
    X = np.asfarray(X,dtype='float')
    #X_pca = PCA(n_components=128).fit_transform(X)
    X_pca = X
    tsne = TSNE(n_components=2, init='random', random_state=0)
    X_r = tsne.fit_transform(X_pca)
    imgPlot(X_r,Y)