From e1616fd376f7a6707273642b8a02769110848038 Mon Sep 17 00:00:00 2001 From: Szymon Bartanowicz Date: Tue, 14 May 2024 23:39:54 +0200 Subject: [PATCH] fix --- model.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/model.py b/model.py index 9e18c49..76d6730 100644 --- a/model.py +++ b/model.py @@ -12,6 +12,8 @@ data = pd.read_csv('./data/train.csv') data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna() data['Age'] = pd.to_numeric(data['Age'], errors='coerce') +data['Age'] = pd.to_numeric(data['BodyweightKg'], errors='coerce') +data['Age'] = pd.to_numeric(data['TotalKg'], errors='coerce') features = data[['Sex', 'Age', 'BodyweightKg']] target = data['TotalKg']