import sys import numpy as np from frovedis.exrpc.server import FrovedisServer from frovedis.mllib.manifold import TSNE desc = "Testing fit_transform for float64 : " # initializing the frovedis server argvs = sys.argv argc = len(argvs) if (argc < 2): print('Please give frovedis_server calling command as the first argument\n\ (e.g. "mpirun -np 2 /opt/nec/frovedis/ve/bin/frovedis_server")') quit() FrovedisServer.initialize(argvs[1]) # sample numpy dense data (3x3) mat = np.matrix( [[0, 0, 0, 0], [0, 1, 1, 1], [1, 0, 1, 0], [1, 1, 1, 0], [1, 1, 1, 1]], dtype=np.float64) tsne = TSNE(n_components=2) try: emb = tsne.fit_transform(mat) print(desc, "Passed") except: print(desc, "Failed") FrovedisServer.shut_down()
# releasing results from server #tsne.embedding_.release() #if mat is FrovedisRowmajorMatrix # sample numpy dense data (3x3) mat1 = np.matrix([[0, 0, 0, 0], [0, 1, 1, 1], [1, 0, 1, 0], [1, 1, 1, 0], [1, 1, 1, 1]], #dtype=np.float32) dtype=np.float64) #mat = FrovedisRowmajorMatrix(mat) # for FrovedisRowmajorMatrix print("input matrix: ") print(mat1) # for numpy matrix #print(mat.debug_print()) # for FrovedisRowmajorMatrix tsne1 = TSNE() embeddings = tsne1.fit_transform(mat1) print("embeddings_: ") print(embeddings) # for numpy matrix #print(embeddings.debug_print()) for FrovedisRowmajorMatrix # releasing results from server #tsne1.embedding_.release() #if mat is FrovedisRowmajorMatrix FrovedisServer.shut_down()