From a597ba4d2af22ca2eddf23a85b42b4d3ae993696 Mon Sep 17 00:00:00 2001
From: Szymon Bartanowicz <szymon@bartanowicz.pl>
Date: Tue, 14 May 2024 23:40:55 +0200
Subject: [PATCH] fix

---
 model.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/model.py b/model.py
index 76d6730..9801336 100644
--- a/model.py
+++ b/model.py
@@ -12,8 +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')
+data['BodyweightKg'] = pd.to_numeric(data['BodyweightKg'], errors='coerce')
+data['TotalKg'] = pd.to_numeric(data['TotalKg'], errors='coerce')
 features = data[['Sex', 'Age', 'BodyweightKg']]
 target = data['TotalKg']