From f426cd914c70e7e3ffa4cdd2db0da18f287f3341 Mon Sep 17 00:00:00 2001 From: ShaaniBel Date: Fri, 26 Sep 2025 16:42:10 -0400 Subject: [PATCH] changed parkinson variables we're working with --- linear-regression-parkinsons.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/linear-regression-parkinsons.py b/linear-regression-parkinsons.py index cf1c005..e95e600 100644 --- a/linear-regression-parkinsons.py +++ b/linear-regression-parkinsons.py @@ -187,9 +187,9 @@ if __name__ == "__main__": assert df.isna().sum().sum() == 0, "There are still some null values." # split the X and Y values - target = 'total_UPDRS' - x = df.drop(columns=[target]) - y = df[target] + feature_columns = [col for col in df.columns if col not in ['motor_UPDRS', 'total_UPDRS', 'subject#']] + x = df[feature_columns] + y = df['motor_UPDRS'] # train / test splitting (80 / 20) n_train = int(0.8 * len(x))