From 0d2aabccbd5bac7413e1989c80994a4f6d5b9426 Mon Sep 17 00:00:00 2001 From: maelstrom Date: Sat, 7 Dec 2024 22:03:18 -0500 Subject: [PATCH] minor fix --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index c82e0ef..6315016 100644 --- a/train.py +++ b/train.py @@ -15,7 +15,7 @@ def main(): labels = [] with open(path / "train.csv", "r") as file: for line in file: - labels.append(line.strip().split(",")) + labels.append(line.strip().split(",")[0]) spark = SparkSession.builder.appName("train").getOrCreate() pipe = SpectrogramPipe(spark)