Changeset 600 for mds-and-trees/mds_plot.py

Timestamp:

08/28/16 15:04:25 (8 years ago)

Author:

oriona

Message:

Number of returned dimensions set to minimum of desired number and the number returned by mds.

File:

• mds-and-trees/mds_plot.py (modified) (1 diff)

mds-and-trees/mds_plot.py

@@ -77 +77 @@
 def compute_mds(distance_matrix, dim):
         embed, evals = cmdscale(distance_matrix)
+
         variances = [np.var(embed[:,i]) for i in range(len(embed[0]))]
-        embed = np.asarray([embed[:,i] for i in range(dim)]).T
-
-        percent_variances = [sum(variances[:i+1])/sum(variances) for i in range(dim)]
+        percent_variances = [sum(variances[:i+1])/sum(variances) for i in range(len(variances))]
         for i,pv in enumerate(percent_variances):
                 print(i+1,"dimension:",pv)
+
+        dim = min(dim, len(embed[0]))
+        embed = np.asarray([embed[:,i] for i in range(dim)]).T
 
         return embed