Mercurial > repos > sauria > create_correlation_matrix

changeset 4:4a76c97c3dd0 draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Uploaded
author sauria
date Thu, 27 Apr 2017 17:32:19 -0400
parents 89009e9b7eb0
children
files correlation_matrix.py
diffstat 1 files changed, 3 insertions(+), 3 deletions(-) [+]
line wrap: on
line diff
--- 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)