This commit is contained in:
Szymon Bartanowicz 2024-05-14 22:29:02 +02:00
parent 14fc6d1120
commit 7e735543b9

View File

@ -51,10 +51,14 @@ import tensorflow as tf
data = pd.read_csv('./data/train.csv') data = pd.read_csv('./data/train.csv')
features_idx = [1, 4, 7] # Sex, Age, BodyweightKg print(data.columns) # Debugging: Print DataFrame columns
target_idx = 25 # TotalKg
data = data.iloc[:, [1, 4, 7, 25]].dropna() # Assuming the relevant columns are at these indexes
features_idx = [1, 4, 7] # Sex, Age, BodyweightKg
target_idx = 24 # TotalKg
# Dropping rows with NaN values from relevant columns
data = data.iloc[:, [1, 4, 7, 24]].dropna()
features = data.iloc[:, features_idx] features = data.iloc[:, features_idx]
target = data.iloc[:, target_idx] target = data.iloc[:, target_idx]