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

Timestamp:

08/30/16 17:08:10 (8 years ago)

Author:

Maciej Komosinski

Message:

Made the amount of jitter a continuous, adjustable parameter

File:

• mds-and-trees/mds_plot.py (modified) (5 diffs)

mds-and-trees/mds_plot.py

@@ -63 +63 @@
     return Y, evals
 
-def rand_jitter(arr):
-        stdev = arr.max() / 100.
-        return arr + np.random.randn(len(arr)) * stdev * 2
+def rand_jitter(arr, jitter):
+        stdev = (arr.max()-arr.min()) / 100. * jitter #dispersion proportional to range
+        return arr + np.random.randn(len(arr)) * stdev
 
 
@@ -91 +91 @@
         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)
+                print("In",i+1,"dimensions:",pv)
 
         dim = min(dim, len(embed[0]))
@@ -107 +107 @@
                 ax = fig.add_subplot(111, projection='3d')
 
-        add_jitter = lambda tab : rand_jitter(tab) if jitter==1 else tab
+        add_jitter = lambda tab : rand_jitter(tab, jitter) if jitter>0 else tab
 
         x_dim = len(coordinates[0])
@@ -120 +120 @@
                 #       print(point)
                 for label, x, y in zip(labels, points[0], points[1]):
-                        #if label not in knownlabels:
-                        #       knownlabels.append(label)
-                        #       colors.append('#ff0000')
                         for key in labelconvert:
                                 if label.startswith(key):
@@ -165 +162 @@
         parser.add_argument('--dim', required=False, help='number of dimensions of the new space')
         parser.add_argument('--sep', required=False, help='separator of the source file')
-        parser.add_argument('--j', required=False, help='for j=1 random jitter is added to the plot')
+        parser.add_argument('--j', required=False, help='for j>0, random jitter is added to points in the plot')
 
         args = parser.parse_args()
         set_value = lambda value, default : default if value == None else value
-        main(args.input, int(set_value(args.dim, 3)), set_value(args.output, ""), int(set_value(args.j, 0)), set_value(args.sep, "\t"))
+        main(args.input, int(set_value(args.dim, 3)), set_value(args.output, ""), float(set_value(args.j, 0)), set_value(args.sep, "\t"))