generate_tsne_2_feature.py 文件源码

def toTsne(train,test,n_component=2,file_name='tf_idf',path='data/'):
    tsne=TSNE(n_components=n_component,random_state=1123,njobs=-1)
    lentrain=train.shape[0]
    X=np.vstack([train,test])
    tsne.fit(X)
    res=tsne.embedding_
    #print res
    pd.to_pickle(res[:lentrain],path+'train_svd_20_tsne_'+str(n_component)+'_'+file_name+'.pkl')
    pd.to_pickle(res[lentrain:],path+'test_svd_20_tsne_'+str(n_component)+'_'+file_name+'.pkl')
    return 'Success'