Improved: Updated column names for the embedded data sets.

AAnzel · Dec 10, 2021 · ec51442 · ec51442
1 parent 32f3a4d
commit ec51442
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Source/common.py b/Source/common.py
@@ -1276,7 +1276,7 @@ def work_with_fasta(data_set_type, folder_path, key_suffix):
     list_of_vectors = vectorize_mags(
         w2v_model, path_fasta=folder_path, end=num_of_fasta_files)
 
-    column_names = ['Dimension ' + str(i) for i in range(VECTOR_SIZE)]
+    column_names = ['dim ' + str(i) for i in range(VECTOR_SIZE)]
     df = pd.DataFrame(list_of_vectors, columns=column_names)
     list_of_dates = create_temporal_column(
         fasta_files, None, None, 'TIMESTAMP')