# HG changeset patch
# User sauria
# Date 1493328739 14400
# Node ID 4a76c97c3dd05324e60047a8a14df4a43671ad0e
# Parent  89009e9b7eb098555b836a1f786647f45ae75c2f
Uploaded

diff -r 89009e9b7eb0 -r 4a76c97c3dd0 correlation_matrix.py
--- a/correlation_matrix.py	Thu Apr 27 17:28:27 2017 -0400
+++ b/correlation_matrix.py	Thu Apr 27 17:32:19 2017 -0400
@@ -8,7 +8,7 @@
 def main():
     parser = generate_parser()
     args = parser.parse_args()
-    data, rnames = load_data(args)
+    data, names = load_data(args)
     corr = find_correlations(data, args)
     save_data(corr, names, args)
 
@@ -19,7 +19,7 @@
     data = []
     if args.column:
         cnames = infile.readline().split()
-        if args.rows:
+        if args.row:
             cnames = cnames[1:]
     temp = infile.readline()
     if args.int:
@@ -41,7 +41,7 @@
     if args.features:
         data = data.T
         names = cnames
-    return data, names, cnames
+    return data, names
 
 def find_correlations(data, args):
     corr = numpy.ones((data.shape[0], data.shape[0]), dtype=numpy.float32)