batuhan-basoglu/Mini-Project-of-Machine-Learning
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

changed parkinson variables we're working with

Browse source
This commit is contained in:
ShaaniBel 2025-09-26 16:42:10 -04:00
parent de5d62ee8e
commit f426cd914c
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
linear-regression-parkinsons.py
View file

@ -187,9 +187,9 @@ if __name__ == "__main__":
assert df.isna().sum().sum() == 0, "There are still some null values." assert df.isna().sum().sum() == 0, "There are still some null values."
# split the X and Y values # split the X and Y values
target = 'total_UPDRS' feature_columns = [col for col in df.columns if col not in ['motor_UPDRS', 'total_UPDRS', 'subject#']]
x = df.drop(columns=[target]) x = df[feature_columns]
y = df[target] y = df['motor_UPDRS']
# train / test splitting (80 / 20) # train / test splitting (80 / 20)
n_train = int(0.8 * len(x)) n_train = int(0.8 * len(x))