fix(lib): Apply mappings before removing headers (#66)

* 🌋🌈 Updated with Glitch

* 📟🖇 Updated with Glitch

Co-authored-by: Glitch (isair-tensorflow-load-csv) <none>

Author: isair

src/loadCsv.ts

@@ -35,20 +35,18 @@ const loadCsv = (filename: string, options: CsvReadOptions) => {
           })
     );
 
+  const mappedData = applyMappings(data, mappings);
+
   const tables: { [key: string]: CsvTable } = {
-    labels: filterColumns(data, labelColumns),
-    features: filterColumns(data, featureColumns),
+    labels: filterColumns(mappedData, labelColumns),
+    features: filterColumns(mappedData, featureColumns),
     testFeatures: [],
     testLabels: [],
   };
 
   tables.labels.shift();
   tables.features.shift();
 
-  for (const key of Object.keys(tables)) {
-    tables[key] = applyMappings(tables[key], mappings);
-  }
-
   if (shouldShuffle) {
     const seed =
       typeof shouldShuffle === 'string' ? shouldShuffle : defaultShuffleSeed;