xgboost

script/preprocess.ts

@@ -2,14 +2,13 @@ import Bun from 'bun';
 import path from 'node:path';
 import { parseArgs } from 'node:util';
 import fs, { mkdirSync } from 'node:fs';
-import { Song } from 'tja-parser';
 import { featurize } from '../preprocess/featurize';
 import { parseTja } from '../preprocess/parse'
 
 const { values } = parseArgs({
     args: Bun.argv,
     options: {
-        outputDir: {
+        workingDir: {
             type: "string"
         },
         dataDir: {
@@ -20,13 +19,13 @@ const { values } = parseArgs({
     allowPositionals: true,
 })
 
-if (!values.dataDir || !values.outputDir) {
-    console.error("--outputDir --dataDir");
+if (!values.dataDir || !values.workingDir) {
+    console.error("--workingDir --dataDir");
     process.exit(1);
 }
 
-const outputDir = values.outputDir ?? '';
-if (!fs.existsSync(outputDir)) mkdirSync(outputDir)
+const workingDir = values.workingDir ?? '';
+if (!fs.existsSync(workingDir)) mkdirSync(workingDir)
 const dataDir = values.dataDir ?? '';
 
 const tjaDir = path.join(dataDir, 'tja');
@@ -61,5 +60,5 @@ for (const file of files) {
     }
 }
 
-const featurePath = path.join(outputDir, 'features.json');
+const featurePath = path.join(workingDir, 'features.json');
 fs.writeFileSync(featurePath, JSON.stringify(features, null, 2), 'utf-8');