def bhtsne(X):
"""Wrapper for the reference lvdmaaten/bhtsne implementation."""
# PCA preprocessing is done elsewhere in the benchmark script
n_iter = -1 # TODO find a way to report the number of iterations
return run_bh_tsne(X,
use_pca=False,
perplexity=args.perplexity,
verbose=args.verbose > 0), n_iter
def main(args):
vectors = numpy.load(args.input)
v2d = bhtsne.run_bh_tsne(vectors,
no_dims=3,
perplexity=50,
theta=0.5,
randseed=-1,
verbose=True,
initial_dims=50,
use_pca=True,
max_iter=1000)
numpy.save(args.output, v2d)
def bhtsne(X):
"""Wrapper for the reference lvdmaaten/bhtsne implementation."""
# PCA preprocessing is done elsewhere in the benchmark script
n_iter = -1 # TODO find a way to report the number of iterations
return run_bh_tsne(X, use_pca=False, perplexity=args.perplexity,
verbose=args.verbose > 0), n_iter