This commit is contained in:
Szymon Bartanowicz 2024-05-14 23:40:55 +02:00
parent e1616fd376
commit a597ba4d2a

View File

@ -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']