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()