Skip to content

Commit

Permalink
Merge pull request #29 from sauln/28
Browse files Browse the repository at this point in the history
fix #28, issues with print in version 2.7
  • Loading branch information
lmcinnes committed Nov 30, 2017
2 parents 34425ea + 7ea9194 commit 6c0c893
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions umap/umap_.py
@@ -1,6 +1,7 @@
# Author: Leland McInnes <leland.mcinnes@gmail.com>
#
# License: BSD 3 clause
from __future__ import print_function
from collections import deque, namedtuple
from warnings import warn

Expand Down Expand Up @@ -452,7 +453,7 @@ def nn_descent(data, n_neighbors, rng_state, max_candidates=50,

for n in range(n_iters):
if verbose:
print("\tnn descent iteration ", n, " / ", n_iters)
print("\t", n, " / ", n_iters)

candidate_neighbors = build_candidates(current_graph, n_vertices,
n_neighbors, max_candidates,
Expand Down Expand Up @@ -1026,7 +1027,7 @@ def optimize_layout(embedding, positive_head, positive_tail,
alpha = initial_alpha * 0.000001

if verbose and i % int(n_edge_samples/10) == 0 :
print("\tembedding iteration ", i, " / ", n_edge_samples)
print("\t", i, " / ", n_edge_samples)

return embedding

Expand Down

0 comments on commit 6c0c893

Please sign in to comment.