From a94fffa76623163209396a62778bff38e78671d4 Mon Sep 17 00:00:00 2001 From: Dawith Lim Date: Wed, 1 Oct 2025 14:27:48 -0400 Subject: [PATCH] labels updated to load all the feature vector elements in the mat files --- pipe/extract.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pipe/extract.py b/pipe/extract.py index 9f7aa4b..8396d0c 100644 --- a/pipe/extract.py +++ b/pipe/extract.py @@ -27,7 +27,8 @@ def extract(spark: SparkSession) -> DataFrame: """ path = Path("/app/workdir") - labels = [] + labels = [DataKind.BB, DataKind.FPS, DataKind.NSD, DataKind.NCNT, + DataKind.SPEC, DataKind.TARGET, DataKind.TREATMENT] with open(path / "train.csv", "r") as file: for line in file: labels.append(line.strip().split(",")[0])